Merge branch 'vinjai/autocurate' of github.com:scaleapi/nucleus-python-client into vinjai/autocurate

Author: Vinjai Vale

nucleus/autocurate.py

@@ -8,16 +8,9 @@
 from nucleus.job import AsyncJob
 
 
-def entropy(name, model_runs, client):
-    assert (
-        len({model_run.dataset_id for model_run in model_runs}) == 1
-    ), f"Model runs have conflicting dataset ids: {model_runs}"
-    # TODO: support multiple model runs
-    assert (
-        len(model_runs) == 1
-    ), "Entropy currently not supported for multiple model runs"
-    model_run_ids = [model_run.model_run_id for model_run in model_runs]
-    dataset_id = model_runs[0].dataset_id
+def entropy(name, model_run, client):
+    model_run_ids = [model_run.model_run_id]
+    dataset_id = model_run.dataset_id
     response = client.make_request(
         payload={"modelRunIds": model_run_ids},
         route=f"autocurate/{dataset_id}/single_model_entropy/{name}",