fix(ux): remove reference to DatasetQuery (#104)

src/datachain/catalog/catalog.py

@@ -2012,7 +2012,7 @@ def query(
                 )
             if proc.returncode == QUERY_SCRIPT_INVALID_LAST_STATEMENT_EXIT_CODE:
                 raise QueryScriptRunError(
-                    "Last line in a script was not an instance of DatasetQuery",
+                    "Last line in a script was not an instance of DataChain",
                     return_code=proc.returncode,
                     output=output,
                 )

tests/func/test_catalog.py

@@ -1025,7 +1025,7 @@ def test_query_last_statement_not_ds_query_instance(mock_popen, cloud_test_catal
     with pytest.raises(QueryScriptRunError) as exc_info:
         catalog.query(query_script)
         assert str(exc_info.value).startswith(
-            "Last line in a script was not an instance of DatasetQuery"
+            "Last line in a script was not an instance of DataChain"
         )
 
 

tests/func/test_query.py

@@ -149,7 +149,7 @@ def test_query_cli_no_dataset_returned(
 
     with pytest.raises(
         QueryScriptRunError,
-        match="Last line in a script was not an instance of DatasetQuery",
+        match="Last line in a script was not an instance of DataChain",
     ):
         query(catalog, str(filepath), "my-dataset", columns=["name"])
 
@@ -159,7 +159,7 @@ def test_query_cli_no_dataset_returned(
     assert latest_job[1] == os.path.basename(filepath)
     assert latest_job[2] == JobStatus.FAILED
     assert latest_job[3] == JobQueryType.PYTHON
-    assert latest_job[4] == "Last line in a script was not an instance of DatasetQuery"
+    assert latest_job[4] == "Last line in a script was not an instance of DataChain"
     assert latest_job[5].find("datachain.error.QueryScriptRunError")