Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
Weisu Yin committed Feb 17, 2023
1 parent f7f2423 commit 28a6167
Show file tree
Hide file tree
Showing 8 changed files with 0 additions and 8 deletions.
1 change: 0 additions & 1 deletion src/autogluon/cloud/data/format_converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}

Expand Down
1 change: 0 additions & 1 deletion src/autogluon/cloud/predictor/cloud_predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@


class MultiModalCloudPredictor(CloudPredictor):

predictor_file_name = "MultiModalCloudPredictor.pkl"

@property
Expand Down
1 change: 0 additions & 1 deletion src/autogluon/cloud/predictor/tabular_cloud_predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@


class TabularCloudPredictor(CloudPredictor):

predictor_file_name = "TabularCloudPredictor.pkl"

@property
Expand Down
1 change: 0 additions & 1 deletion src/autogluon/cloud/scripts/script_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down
1 change: 0 additions & 1 deletion src/autogluon/cloud/scripts/tabular_serve.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
1 change: 0 additions & 1 deletion src/autogluon/cloud/utils/deserializers.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,6 @@ def deserialize(self, stream) -> pd.DataFrame:


class PandasDeserializeStrategyFactory:

__supported_strategy = [
ParquetPandasDeserializeStrategy,
CSVPandasDeserializeStrategy,
Expand Down
1 change: 0 additions & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit 28a6167

Please sign in to comment.