Deprecate get_state and remove deprecations

composer/callbacks/mlperf.py

@@ -263,22 +263,8 @@ def _get_dataloader_stats(self, dataloader: Iterable):
             if isinstance(dataloader.dataset, IterableDataset):
                 num_samples *= dist.get_world_size()
             return (dataloader.batch_size, num_samples)
-        try:
-            # attempt to import ffcv and test if its an ffcv loader.
-            import ffcv  # type: ignore
-
-            warnings.warn(DeprecationWarning('ffcv is deprecated and will be removed in v0.18'))
-
-            if isinstance(dataloader, ffcv.loader.Loader):
-                # Use the cached attribute ffcv.init_traversal_order to compute number of samples
-                return (
-                    dataloader.batch_size,  # type: ignore
-                    len(dataloader.next_traversal_order()) * dist.get_world_size()  # type: ignore
-                )
-        except ImportError:
-            pass
-
-        raise TypeError(f'torch dataloader or ffcv dataloader required (and ffcv installed)')
+
+        raise TypeError(f'torch dataloader required')
 
     def fit_start(self, state: State, logger: Logger) -> None:
         if _global_rank_zero():

composer/core/state.py

@@ -733,7 +733,8 @@ def fsdp_sharded_state_dict_enabled(self):
 
     @property
     def fsdp_elastic_sharded_enabled(self):
-        warnings.warn('state.fsdp_elastic_sharded_enabled is deprecated and will be removed v0.21.0')
+        warnings.warn('state.fsdp_elastic_sharded_enabled is deprecated and will be removed v0.21.0',
+                      DeprecationWarning)
         return self.fsdp_sharded_state_dict_enabled
 
     @property