Bump black from 23.10.1 to 24.3.0

custom_model_runner/datarobot_drum/drum/adapters/cli/drum_fit_adapter.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import json
 import logging
 import shutil

custom_model_runner/datarobot_drum/drum/adapters/cli/drum_score_adapter.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 
 from datarobot_drum.drum.adapters.cli.shared.drum_class_label_adapter import DrumClassLabelAdapter

custom_model_runner/datarobot_drum/drum/adapters/cli/shared/drum_class_label_adapter.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import sys
 

custom_model_runner/datarobot_drum/drum/adapters/cli/shared/drum_input_file_adapter.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import sys
 from typing import Optional

custom_model_runner/datarobot_drum/drum/adapters/model_adapters/python_model_adapter.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 import sys
@@ -440,9 +441,9 @@ def supported_payload_formats(self):
 
     def model_info(self):
         return {
-            ModelInfoKeys.PREDICTOR: None
-            if self._predictor_to_use is None
-            else self._predictor_to_use.name,
+            ModelInfoKeys.PREDICTOR: (
+                None if self._predictor_to_use is None else self._predictor_to_use.name
+            ),
         }
 
     def load_data(self, binary_data, mimetype, try_hook=True, sparse_colnames=None):

custom_model_runner/datarobot_drum/drum/adapters/model_adapters/r_model_adapter.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 from typing import Optional

custom_model_runner/datarobot_drum/drum/args_parser.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import argparse
 import os
 import subprocess

custom_model_runner/datarobot_drum/drum/artifact_predictors/artifact_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 from abc import ABC, abstractmethod
 import logging
 from typing import Optional

custom_model_runner/datarobot_drum/drum/artifact_predictors/sklearn_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import pickle
 
 from datarobot_drum.drum.artifact_predictors.artifact_predictor import ArtifactPredictor

custom_model_runner/datarobot_drum/drum/artifact_predictors/torch_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import sys
 
 import numpy as np

custom_model_runner/datarobot_drum/drum/artifact_predictors/xgboost_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import pickle
 
 from datarobot_drum.drum.artifact_predictors.artifact_predictor import ArtifactPredictor

custom_model_runner/datarobot_drum/drum/common.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 import sys

custom_model_runner/datarobot_drum/drum/data_marshalling.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 from distutils.util import strtobool
 from typing import Any, List, Optional, Union

custom_model_runner/datarobot_drum/drum/description.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 version = "1.10.20"
 __version__ = version
 project_name = "datarobot-drum"

custom_model_runner/datarobot_drum/drum/drum.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import copy
 import glob
 import json
@@ -740,12 +741,16 @@ def _prepare_prediction_server_or_batch_pipeline(self, run_language):
             "triton_grpc_port": options.triton_grpc_port,
             "api_token": options.api_token,
             "allow_dr_api_access": options.allow_dr_api_access,
-            "query_params": '"{}"'.format(options.query)
-            if getattr(options, "query", None) is not None
-            else "null",
-            "content_type": '"{}"'.format(options.content_type)
-            if getattr(options, "content_type", None) is not None
-            else "null",
+            "query_params": (
+                '"{}"'.format(options.query)
+                if getattr(options, "query", None) is not None
+                else "null"
+            ),
+            "content_type": (
+                '"{}"'.format(options.content_type)
+                if getattr(options, "content_type", None) is not None
+                else "null"
+            ),
             "target_type": self.target_type.value,
             "user_secrets_mount_path": getattr(options, "user_secrets_mount_path", None),
             "user_secrets_prefix": getattr(options, "user_secrets_prefix", None),
@@ -769,16 +774,18 @@ def _prepare_prediction_server_or_batch_pipeline(self, run_language):
                     "port": port,
                     "show_perf": str(options.show_perf).lower(),
                     "engine_type": "RestModelServing" if options.production else "Generic",
-                    "component_type": "uwsgi_serving"
-                    if options.production
-                    else "prediction_server",
-                    "uwsgi_max_workers": options.max_workers
-                    if getattr(options, "max_workers")
-                    else "null",
+                    "component_type": (
+                        "uwsgi_serving" if options.production else "prediction_server"
+                    ),
+                    "uwsgi_max_workers": (
+                        options.max_workers if getattr(options, "max_workers") else "null"
+                    ),
                     "single_uwsgi_worker": (options.max_workers == 1),
-                    "deployment_config": '"{}"'.format(options.deployment_config)
-                    if getattr(options, "deployment_config", None) is not None
-                    else "null",
+                    "deployment_config": (
+                        '"{}"'.format(options.deployment_config)
+                        if getattr(options, "deployment_config", None) is not None
+                        else "null"
+                    ),
                 }
             )
 

custom_model_runner/datarobot_drum/drum/enum.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 from enum import Enum

custom_model_runner/datarobot_drum/drum/language_predictors/base_language_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 import time

custom_model_runner/datarobot_drum/drum/language_predictors/java_predictor/java_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import glob
 import logging
 import os

custom_model_runner/datarobot_drum/drum/language_predictors/julia_predictor/julia_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 

custom_model_runner/datarobot_drum/drum/language_predictors/python_predictor/python_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import tempfile
 import shutil

custom_model_runner/datarobot_drum/drum/language_predictors/r_predictor/r_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 
@@ -260,9 +261,11 @@ def _transform(self, **kwargs):
         with capture_R_traceback_if_errors(r_handler, logger):
             transformations = r_handler.outer_transform(
                 binary_data=ro.vectors.ByteVector(input_binary_data),
-                target_binary_data=ro.NULL
-                if target_binary_data is None
-                else ro.vectors.ByteVector(target_binary_data),
+                target_binary_data=(
+                    ro.NULL
+                    if target_binary_data is None
+                    else ro.vectors.ByteVector(target_binary_data)
+                ),
                 mimetype=ro.NULL if mimetype is None else mimetype,
                 transformer=self._model,
                 sparse_colnames=self._get_sparse_colnames(kwargs),

custom_model_runner/datarobot_drum/drum/language_predictors/triton_predictor/triton_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 
 import requests

custom_model_runner/datarobot_drum/drum/model_metadata.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import json
 import os
 import re

custom_model_runner/datarobot_drum/drum/perf_testing.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import collections
 import json
 import os

custom_model_runner/datarobot_drum/drum/push.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import os
 import re
 from pathlib import Path

custom_model_runner/datarobot_drum/drum/resource_monitor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import os
 import psutil
 

custom_model_runner/datarobot_drum/drum/runtime.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 from datarobot_drum.drum.server import (
     empty_api_blueprint,

custom_model_runner/datarobot_drum/drum/server.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 from flask import Flask, Blueprint
 import os
 

custom_model_runner/datarobot_drum/drum/templates_generator.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import os
 import logging
 import time

custom_model_runner/datarobot_drum/drum/typeschema_validation.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import numbers
 import sys
 from abc import ABC, abstractmethod

custom_model_runner/datarobot_drum/drum/utils/dataframe.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 from typing import List, Tuple, Optional, Union
 
 import numpy as np

custom_model_runner/datarobot_drum/drum/utils/drum_utils.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import glob
 import json
 import logging

custom_model_runner/datarobot_drum/drum/utils/stacktraces.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 from contextlib import contextmanager
 

custom_model_runner/datarobot_drum/drum/utils/structured_input_read_utils.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import io
 import logging
 import os

custom_model_runner/datarobot_drum/profiler/stats_collector.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import pandas as pd
 import time
 

custom_model_runner/datarobot_drum/resource/components/Python/generic_predictor/generic_predictor.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import urllib
 from typing import Optional

custom_model_runner/datarobot_drum/resource/components/Python/prediction_server/prediction_server.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import sys
 from pathlib import Path

custom_model_runner/datarobot_drum/resource/components/Python/uwsgi_component/uwsgi_serving.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 import sys

custom_model_runner/datarobot_drum/resource/deployment_config_helpers.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import json
 
 from datarobot_drum.drum.exceptions import DrumCommonException

custom_model_runner/datarobot_drum/resource/drum_server_utils.py

@@ -4,6 +4,7 @@
 This is proprietary source code of DataRobot, Inc. and its affiliates.
 Released under the terms of DataRobot Tool and Utility Agreement.
 """
+
 import logging
 import os
 from typing import Optional