diff --git a/src/autogluon/cloud/data/format_converter.py b/src/autogluon/cloud/data/format_converter.py index 34200bf..b6c695f 100644 --- a/src/autogluon/cloud/data/format_converter.py +++ b/src/autogluon/cloud/data/format_converter.py @@ -117,7 +117,6 @@ def _save_dataframe(self, data: pd.DataFrame, path: str): class FormatConverterFactory: - __supported_converters = [CSVConverter, ParquetConverter] __ext_to_converter = {cls().ext: cls for cls in __supported_converters} diff --git a/src/autogluon/cloud/predictor/cloud_predictor.py b/src/autogluon/cloud/predictor/cloud_predictor.py index bce8ffd..2580aeb 100644 --- a/src/autogluon/cloud/predictor/cloud_predictor.py +++ b/src/autogluon/cloud/predictor/cloud_predictor.py @@ -59,7 +59,6 @@ class CloudPredictor(ABC): - predictor_file_name = "CloudPredictor.pkl" def __init__(self, cloud_output_path: str, local_output_path: Optional[str] = None, verbosity: int = 2) -> None: diff --git a/src/autogluon/cloud/predictor/multimodal_cloud_predictor.py b/src/autogluon/cloud/predictor/multimodal_cloud_predictor.py index 37437ed..750ab4f 100644 --- a/src/autogluon/cloud/predictor/multimodal_cloud_predictor.py +++ b/src/autogluon/cloud/predictor/multimodal_cloud_predictor.py @@ -14,7 +14,6 @@ class MultiModalCloudPredictor(CloudPredictor): - predictor_file_name = "MultiModalCloudPredictor.pkl" @property diff --git a/src/autogluon/cloud/predictor/tabular_cloud_predictor.py b/src/autogluon/cloud/predictor/tabular_cloud_predictor.py index 1c716b4..6e2274d 100644 --- a/src/autogluon/cloud/predictor/tabular_cloud_predictor.py +++ b/src/autogluon/cloud/predictor/tabular_cloud_predictor.py @@ -15,7 +15,6 @@ class TabularCloudPredictor(CloudPredictor): - predictor_file_name = "TabularCloudPredictor.pkl" @property diff --git a/src/autogluon/cloud/scripts/script_manager.py b/src/autogluon/cloud/scripts/script_manager.py index a2bc98e..7ee9123 100644 --- a/src/autogluon/cloud/scripts/script_manager.py +++ b/src/autogluon/cloud/scripts/script_manager.py @@ -3,7 +3,6 @@ class ScriptManager: - CLOUD_PATH = Path(__file__).parent.parent.absolute() SCRIPTS_PATH = os.path.join(CLOUD_PATH, "scripts") TRAIN_SCRIPT_PATH = os.path.join(SCRIPTS_PATH, "train.py") diff --git a/src/autogluon/cloud/scripts/tabular_serve.py b/src/autogluon/cloud/scripts/tabular_serve.py index 5391469..78074cd 100644 --- a/src/autogluon/cloud/scripts/tabular_serve.py +++ b/src/autogluon/cloud/scripts/tabular_serve.py @@ -37,7 +37,6 @@ def model_fn(model_dir): def transform_fn(model, request_body, input_content_type, output_content_type="application/json"): - if input_content_type == "application/x-parquet": buf = BytesIO(request_body) data = pd.read_parquet(buf) diff --git a/src/autogluon/cloud/utils/deserializers.py b/src/autogluon/cloud/utils/deserializers.py index e7dc4f2..4d3458c 100644 --- a/src/autogluon/cloud/utils/deserializers.py +++ b/src/autogluon/cloud/utils/deserializers.py @@ -53,7 +53,6 @@ def deserialize(self, stream) -> pd.DataFrame: class PandasDeserializeStrategyFactory: - __supported_strategy = [ ParquetPandasDeserializeStrategy, CSVPandasDeserializeStrategy, diff --git a/tests/conftest.py b/tests/conftest.py index 52bd6eb..27a2720 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -8,7 +8,6 @@ class CloudTestHelper: - cpu_training_image = "369469875935.dkr.ecr.us-east-1.amazonaws.com/autogluon-nightly-training:cpu-latest" gpu_training_image = "369469875935.dkr.ecr.us-east-1.amazonaws.com/autogluon-nightly-training:gpu-latest" cpu_inference_image = "369469875935.dkr.ecr.us-east-1.amazonaws.com/autogluon-nightly-inference:cpu-latest"