Merge branch 'mixin/pt-init' of https://github.com/Lightning-AI/models into mixin/pt-init

Author: Borda

src/litmodels/integrations/mixins.py

@@ -148,7 +148,7 @@ def push_to_registry(
             except Exception as e:
                 raise RuntimeError(
                     f"Failed to save model arguments: {e}."
-                    " Ensure the models arguments are JSON serializable or user `PickleRegistryMixin`."
+                    " Ensure the model's arguments are JSON serializable or use `PickleRegistryMixin`."
                 ) from e
         elif not hasattr(self, "__init_kwargs"):
             rank_zero_warn(
@@ -160,7 +160,7 @@ def push_to_registry(
         torch_state_dict_path = Path(temp_folder) / f"{model_name}.pth"
         torch.save(self.state_dict(), torch_state_dict_path)
         model_registry = f"{name}:{version}" if version else name
-        # todo: consider created another temp folder and copy there these two files
+        # todo: consider creating another temp folder and copying these two files
         upload_model_files(name=model_registry, path=temp_folder)
 
     @classmethod

tests/integrations/test_cloud.py

@@ -244,7 +244,7 @@ def test_pickle_mixin_push_and_pull():
 
 # This is a dummy model for PyTorch that uses the PyTorchRegistryMixin.
 # This mixin has to be first in the inheritance order.
-# Otherwise, `PyTorchRegistryMixin.__init__` nee to be called explicitly.
+# Otherwise, `PyTorchRegistryMixin.__init__` need to be called explicitly.
 class DummyTorchModel(PyTorchRegistryMixin, torch.nn.Module):
     def __init__(self, input_size: int, output_size: int = 10):
         # PyTorchRegistryMixin.__init__ will capture these arguments