fix

Author: Borda

examples/train-resume.py

@@ -8,7 +8,7 @@
     dataset = tv.datasets.MNIST(".", download=True, transform=tv.transforms.ToTensor())
     train, val = data.random_split(dataset, [55000, 5000])
 
-    model_path = download_model(name="jirka/kaggle/lit-auto-encoder-simple", download_dir="my_models")
+    model_path = download_model(name="jirka/kaggle/lit-auto-encoder-simple:latest", download_dir="my_models")
     print(f"model: {model_path}")
     # autoencoder = LitAutoEncoder.load_from_checkpoint(checkpoint_path=model_path)
 

tests/test_cloud_io.py tests/test_io_cloud.pytests/test_cloud_io.py renamed to tests/test_io_cloud.py

@@ -30,7 +30,7 @@ def test_upload_model(mocker, tmpdir, model, model_path):
 
     # The lit-logger function is just a wrapper around the SDK function
     upload_model(
-        model,
+        model=model,
         name="org-name/teamspace/model-name",
         cluster_id="cluster_id",
         staging_dir=tmpdir,