[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/litmodels/integrations/mixins.py

@@ -26,9 +26,7 @@ def push_to_registry(
         """
 
     @classmethod
-    def pull_from_registry(
-        cls, name: str, version: Optional[str] = None, temp_folder: Optional[str] = None
-    ) -> object:
+    def pull_from_registry(cls, name: str, version: Optional[str] = None, temp_folder: Optional[str] = None) -> object:
         """Pull the model from the registry.
 
         Args:
@@ -67,9 +65,7 @@ def push_to_registry(
         upload_model(name=name, model=pickle_path)
 
     @classmethod
-    def pull_from_registry(
-        cls, name: str, version: Optional[str] = None, temp_folder: Optional[str] = None
-    ) -> object:
+    def pull_from_registry(cls, name: str, version: Optional[str] = None, temp_folder: Optional[str] = None) -> object:
         """Pull the model from the registry.
 
         Args:

tests/integrations/test_cloud.py

@@ -238,4 +238,4 @@ def test_picklemixin_push_and_pull():
     assert loaded_dummy.value == 42
 
     # CLEANING
-    _cleanup_model(teamspace, model_name, expected_num_versions=1)
+    _cleanup_model(teamspace, model_name, expected_num_versions=1)

tests/integrations/test_mixins.py

@@ -26,9 +26,7 @@ def test_pickle_push_and_pull(mock_download_model, mock_upload_model, tmp_path):
     # Set the mock to return the full path to the pickle file.
     mock_download_model.return_value = ["DummyModel.pkl"]
     # Call pull_from_registry and load the DummyModel instance.
-    loaded_dummy = DummyModel.pull_from_registry(
-        name="dummy_model", version="v1", temp_folder=str(tmp_path)
-    )
+    loaded_dummy = DummyModel.pull_from_registry(name="dummy_model", version="v1", temp_folder=str(tmp_path))
     # Verify that the unpickled instance has the expected value.
     assert loaded_dummy.value == 42