✅ Add a test for tracking with params

lamin_cli/_save.py

@@ -53,6 +53,7 @@ def get_stem_uid_and_version_from_file(
             "\nCall `ln.track()` and copy/paste the output"
             " into the notebook"
         )
+    logger.important(f"mapped '{file_path}' on uid '{uid}'")
     return uid, stem_uid, version
 
 

tests/scripts/run-track-with-params.py

@@ -0,0 +1,35 @@
+import argparse
+import lamindb as ln
+
+
+# this section should typically be outside of the script
+ln.Param(name="dataset_key", dtype="str").save()
+ln.Param(name="learning_rate", dtype="float").save()
+ln.Param(name="downsample", dtype="bool").save()
+ln.Param(name="split", dtype="str").save()
+
+
+def main():
+    parser = argparse.ArgumentParser(description="A demo script.")
+    parser.add_argument("--dataset-key", type=str)
+    parser.add_argument("--downsample", action="store_true", help="Downsample")
+    parser.add_argument("--split", choices=["train", "test", "validation"])
+    parser.add_argument("--learning-rate", type=float)
+    args = parser.parse_args()
+
+    params = {
+        "dataset_key": args.dataset_key,
+        "learning_rate": args.learning_rate,
+        "downsample": args.downsample,
+        "split": args.split,
+    }
+
+    ln.track("JjRF4mACd9m00000", params=params)
+
+    # actually do stuff
+
+    ln.finish()
+
+
+if __name__ == "__main__":
+    main()

tests/test_save_scripts.py

@@ -131,3 +131,37 @@ def test_run_save_cache():
     )
     print(result.stderr.decode())
     assert result.returncode == 0
+
+
+def test_run_save_with_params():
+    env = os.environ
+    env["LAMIN_TESTING"] = "true"
+    filepath = scripts_dir / "run-track-with-params.py"
+
+    # run the script
+    result = subprocess.run(
+        f"python {filepath} --dataset-key mydata --learning-rate 0.01 --split train",
+        shell=True,
+        capture_output=True,
+    )
+    print(result.stdout.decode())
+    print(result.stderr.decode())
+    assert result.returncode == 0
+    assert "created Transform" in result.stdout.decode()
+    assert "JjRF4mACd9m00000" in result.stdout.decode()
+    assert "created Run" in result.stdout.decode()
+
+    # you can re-save the script
+    result = subprocess.run(
+        f"lamin save {filepath}",
+        shell=True,
+        capture_output=True,
+        env=env,
+    )
+    print(result.stdout.decode())
+    print(result.stderr.decode())
+    assert result.returncode == 0
+    assert "source code is already saved" in result.stdout.decode()
+    assert (
+        "run-track-with-params.py' on uid 'JjRF4mACd9m00000'" in result.stdout.decode()
+    )