lints and bump

opsml/helpers/gcp_utils.py

@@ -72,7 +72,10 @@ def create_gcp_creds_from_base64(self, service_base64_creds: str) -> Tuple[Crede
         """
         scopes = {"scopes": ["https://www.googleapis.com/auth/devstorage.full_control"]}  # needed for gcsfs
         key = self.decode_base64(service_base64_creds=service_base64_creds)
-        service_creds: Credentials = service_account.Credentials.from_service_account_info(info=key, **scopes)  # type: ignore # noqa
+        service_creds: Credentials = service_account.Credentials.from_service_account_info(  # type: ignore # noqa
+            info=key,
+            **scopes,
+        )
         project_name = cast(str, service_creds.project_id)
 
         return service_creds, project_name, False

opsml/registry/sql/connectors/base.py

@@ -108,7 +108,7 @@ def __init__(self, tracking_uri: str, credentials: Any = None):
     @property
     def _ip_type(self) -> Enum:
         """Sets IP type for CloudSql"""
-        from google.cloud.sql.connector.instance import IPTypes
+        from google.cloud.sql.connector import IPTypes
 
         type_ = IPTypes.PRIVATE if os.environ.get("PRIVATE_IP") else IPTypes.PUBLIC
 

pyproject.toml

@@ -36,7 +36,7 @@ Source = "https://github.com/demml/opsml"
 
 [tool.poetry]
 name = "opsml"
-version = "2.2.0"
+version = "2.3.0"
 readme = "README.md"
 description = "Python MLOPs quality control tooling for your production ML workflows"
 authors = [