def __init__()

in use-cases/inferencing/batch-inference/src/run_batch_predictions.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self):  # Constructor
        self.api_endpoint = os.environ["ENDPOINT"]
        self.model_name = os.environ["MODEL_PATH"]
        self.output_file = os.environ["PREDICTIONS_FILE"]
        self.gcs_bucket = os.environ["BUCKET"]
        self.dataset_output_path = os.environ["DATASET_OUTPUT_PATH"]
        test_dataset = load_from_disk(
            f"gs://{self.gcs_bucket}/{self.dataset_output_path}/test"
        )
        self.test_df = test_dataset.to_pandas()
        # Concatenate vertically (stack rows)
        self.df = pd.concat([self.test_df], axis=0)
        self.df.reset_index(drop=True, inplace=True)