diff --git a/submarine-sdk/pysubmarine/submarine/experiment/api_client.py b/submarine-sdk/pysubmarine/submarine/experiment/api_client.py index 7720f65a42..5bbc5c0e81 100644 --- a/submarine-sdk/pysubmarine/submarine/experiment/api_client.py +++ b/submarine-sdk/pysubmarine/submarine/experiment/api_client.py @@ -37,10 +37,10 @@ # python 2 and python 3 compatibility library import six +from dateutil.parser import parse from six.moves.urllib.parse import quote import submarine.experiment.models -from dateutil.parser import parse from submarine.experiment import rest from submarine.experiment.configuration import Configuration from submarine.experiment.exceptions import ApiException, ApiValueError diff --git a/submarine-sdk/pysubmarine/submarine/store/sqlalchemy_store.py b/submarine-sdk/pysubmarine/submarine/store/sqlalchemy_store.py index f08d038d3d..38c5a175a1 100644 --- a/submarine-sdk/pysubmarine/submarine/store/sqlalchemy_store.py +++ b/submarine-sdk/pysubmarine/submarine/store/sqlalchemy_store.py @@ -18,6 +18,7 @@ from contextlib import contextmanager import sqlalchemy + from submarine.exceptions import SubmarineException from submarine.store.abstract_store import AbstractStore from submarine.store.database.models import Base, SqlMetric, SqlParam diff --git a/submarine-sdk/pysubmarine/submarine/utils/tf_utils.py b/submarine-sdk/pysubmarine/submarine/utils/tf_utils.py index 545395891c..96da0ce1f7 100644 --- a/submarine-sdk/pysubmarine/submarine/utils/tf_utils.py +++ b/submarine-sdk/pysubmarine/submarine/utils/tf_utils.py @@ -17,6 +17,7 @@ import os import tensorflow as tf + from submarine.ml.tensorflow.optimizer import get_optimizer diff --git a/submarine-sdk/pysubmarine/tests/tracking/test_utils.py b/submarine-sdk/pysubmarine/tests/tracking/test_utils.py index 3348fcb7a0..33422b9047 100644 --- a/submarine-sdk/pysubmarine/tests/tracking/test_utils.py +++ b/submarine-sdk/pysubmarine/tests/tracking/test_utils.py @@ -16,6 +16,7 @@ import os import mock + from submarine.store import DEFAULT_SUBMARINE_JDBC_URL from submarine.store.sqlalchemy_store import SqlAlchemyStore from submarine.tracking.utils import (_JOB_NAME_ENV_VAR, _TRACKING_URI_ENV_VAR, diff --git a/submarine-sdk/pysubmarine/tests/utils/test_env.py b/submarine-sdk/pysubmarine/tests/utils/test_env.py index 7aa2b8f164..eefd176e9d 100644 --- a/submarine-sdk/pysubmarine/tests/utils/test_env.py +++ b/submarine-sdk/pysubmarine/tests/utils/test_env.py @@ -17,6 +17,7 @@ from os import environ import pytest + from submarine.utils.env import (get_env, get_from_dicts, get_from_json, get_from_registry, unset_variable) diff --git a/submarine-sdk/pysubmarine/tests/utils/test_rest_utils.py b/submarine-sdk/pysubmarine/tests/utils/test_rest_utils.py index c5e6f063a9..3a866e93a3 100644 --- a/submarine-sdk/pysubmarine/tests/utils/test_rest_utils.py +++ b/submarine-sdk/pysubmarine/tests/utils/test_rest_utils.py @@ -17,6 +17,7 @@ import pytest from mock import Mock, patch + from submarine.exceptions import RestException, SubmarineException from submarine.utils.rest_utils import http_request, verify_rest_response diff --git a/submarine-sdk/pysubmarine/tests/utils/test_tf_utils.py b/submarine-sdk/pysubmarine/tests/utils/test_tf_utils.py index 86675079b0..52ed4ad097 100644 --- a/submarine-sdk/pysubmarine/tests/utils/test_tf_utils.py +++ b/submarine-sdk/pysubmarine/tests/utils/test_tf_utils.py @@ -14,6 +14,7 @@ # limitations under the License. import pytest + from submarine.utils.tf_utils import get_tf_config diff --git a/submarine-sdk/pysubmarine/tests/utils/test_validation.py b/submarine-sdk/pysubmarine/tests/utils/test_validation.py index 285215c2a5..0030913d2e 100644 --- a/submarine-sdk/pysubmarine/tests/utils/test_validation.py +++ b/submarine-sdk/pysubmarine/tests/utils/test_validation.py @@ -14,6 +14,7 @@ # limitations under the License. import pytest + from submarine.exceptions import SubmarineException from submarine.utils.validation import (_validate_db_type_string, _validate_length_limit,