Remove MosaicMLLambdaEvalClient (mosaicml#3432)

mvpatel2000 · Jul 21, 2024 · 9dbdafe · 9dbdafe
1 parent c6a8546
commit 9dbdafe
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 128 deletions.
diff --git a/composer/utils/__init__.py b/composer/utils/__init__.py
@@ -30,7 +30,7 @@
     is_compressed_pt,
 )
 from composer.utils.device import get_device, is_hpu_installed, is_xla_installed
-from composer.utils.eval_client import EvalClient, LambdaEvalClient, LocalEvalClient, MosaicMLLambdaEvalClient
+from composer.utils.eval_client import EvalClient, LambdaEvalClient, LocalEvalClient
 from composer.utils.file_helpers import (
     FORMAT_NAME_WITH_DIST_AND_TIME_TABLE,
     FORMAT_NAME_WITH_DIST_TABLE,
@@ -140,7 +140,6 @@
     'EvalClient',
     'LambdaEvalClient',
     'LocalEvalClient',
-    'MosaicMLLambdaEvalClient',
     'partial_format',
     'add_vision_dataset_transform',
     'VersionedDeprecationWarning',

diff --git a/composer/utils/eval_client/__init__.py b/composer/utils/eval_client/__init__.py
@@ -6,11 +6,9 @@
 from composer.utils.eval_client.eval_client import EvalClient
 from composer.utils.eval_client.lambda_eval_client import LambdaEvalClient
 from composer.utils.eval_client.local_eval_client import LocalEvalClient
-from composer.utils.eval_client.mosaicml_lambda_eval_client import MosaicMLLambdaEvalClient
 
 __all__ = [
     'EvalClient',
     'LambdaEvalClient',
     'LocalEvalClient',
-    'MosaicMLLambdaEvalClient',
 ]
diff --git a/composer/utils/eval_client/mosaicml_lambda_eval_client.py b/composer/utils/eval_client/mosaicml_lambda_eval_client.py
diff --git a/tests/utils/eval_client/test_mcli_eval_client.py b/tests/utils/eval_client/test_mcli_eval_client.py