discovery-engine-api 0.1.34__tar.gz → 0.1.37__tar.gz

{discovery_engine_api-0.1.34 → discovery_engine_api-0.1.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: discovery-engine-api
-Version: 0.1.34
+Version: 0.1.37
 Summary: Python SDK for the Discovery Engine API
 Project-URL: Homepage, https://github.com/leap-laboratories/discovery
 Project-URL: Documentation, https://github.com/leap-laboratories/discovery

{discovery_engine_api-0.1.34 → discovery_engine_api-0.1.37}/discovery/__init__.py

@@ -1,6 +1,6 @@
 """Discovery Engine Python SDK."""
 
-__version__ = "0.1.34"
+__version__ = "0.1.37"
 
 from discovery.client import Engine
 from discovery.types import (

{discovery_engine_api-0.1.34 → discovery_engine_api-0.1.37}/discovery/client.py

@@ -296,7 +296,7 @@ class Engine:
         dataset_id: str,
         target_column_id: str,
         task: str = "regression",
-        mode: str = "fast",
+        depth_iterations: int = 1,
         visibility: str = "public",
         timeseries_groups: Optional[List[Dict[str, Any]]] = None,
         target_column_override: Optional[str] = None,
@@ -311,7 +311,7 @@ class Engine:
             dataset_id: Dataset ID
             target_column_id: Target column ID
             task: Task type (regression, binary_classification, multiclass_classification)
-            mode: Analysis mode ("fast" or "deep")
+            depth_iterations: Number of iterative feature removal cycles (1 = fastest)
             visibility: Dataset visibility ("public" or "private")
             timeseries_groups: Optional list of timeseries column groups
             target_column_override: Optional override for target column name
@@ -327,7 +327,7 @@ class Engine:
         payload = {
             "run_target_column_id": target_column_id,
             "task": task,
-            "mode": mode,
+            "depth_iterations": depth_iterations,
             "visibility": visibility,
             "auto_report_use_llm_evals": auto_report_use_llm_evals,
         }
@@ -458,7 +458,7 @@ class Engine:
         self,
         file: Union[str, Path, "pd.DataFrame"],
         target_column: str,
-        mode: str = "fast",
+        depth_iterations: int = 1,
         title: Optional[str] = None,
         description: Optional[str] = None,
         column_descriptions: Optional[Dict[str, str]] = None,
@@ -483,7 +483,7 @@ class Engine:
         Args:
             file: File path, Path object, or pandas DataFrame
             target_column: Name of the target column
-            mode: Analysis mode ("fast" or "deep", default: "fast")
+            depth_iterations: Number of iterative feature removal cycles (1 = fastest)
             title: Optional dataset title
             description: Optional dataset description
             column_descriptions: Optional dict mapping column names to descriptions
@@ -535,7 +535,7 @@ class Engine:
         files = {"file": (filename, file_content, mime_type)}
         data: Dict[str, Any] = {
             "target_column": target_column,
-            "mode": mode,
+            "depth_iterations": str(depth_iterations),
             "visibility": visibility,
         }
 
@@ -553,7 +553,9 @@ class Engine:
             data["timeseries_groups"] = json.dumps(timeseries_groups)
 
         # Call dashboard API to create report
-        print(f"🚀 Uploading file and creating run (mode: {mode}, target: {target_column})...")
+        print(
+            f"🚀 Uploading file and creating run (depth: {depth_iterations}, target: {target_column})..."
+        )
         # httpx automatically handles multipart/form-data when both files and data are provided
         response = await client.post("/api/reports/create", files=files, data=data)
         response.raise_for_status()
@@ -607,7 +609,7 @@ class Engine:
         self,
         file: Union[str, Path, "pd.DataFrame"],
         target_column: str,
-        mode: str = "fast",
+        depth_iterations: int = 1,
         title: Optional[str] = None,
         description: Optional[str] = None,
         column_descriptions: Optional[Dict[str, str]] = None,
@@ -631,7 +633,7 @@ class Engine:
         Args:
             file: File path, Path object, or pandas DataFrame
             target_column: Name of the target column
-            mode: Analysis mode ("fast" or "deep", default: "fast")
+            depth_iterations: Number of iterative feature removal cycles (1 = fastest)
             title: Optional dataset title
             description: Optional dataset description
             column_descriptions: Optional dict mapping column names to descriptions
@@ -653,7 +655,7 @@ class Engine:
         coro = self.run_async(
             file,
             target_column,
-            mode,
+            depth_iterations,
             title=title,
             description=description,
             column_descriptions=column_descriptions,

{discovery_engine_api-0.1.34 → discovery_engine_api-0.1.37}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "discovery-engine-api"
-version = "0.1.34"
+version = "0.1.37"
 description = "Python SDK for the Discovery Engine API"
 readme = "README.md"
 requires-python = ">=3.10"

{discovery_engine_api-0.1.34 → discovery_engine_api-0.1.37}/tests/test_client.py

@@ -463,7 +463,7 @@ class TestCreateRun:
                 dataset_id=dataset_id,
                 target_column_id=target_column_id,
                 task="regression",
-                mode="fast",
+                depth_iterations=1,
             )
 
         assert result == sample_run
@@ -472,7 +472,7 @@ class TestCreateRun:
         payload = call_args[1]["json"]
         assert payload["run_target_column_id"] == target_column_id
         assert payload["task"] == "regression"
-        assert payload["mode"] == "fast"
+        assert payload["depth_iterations"] == 1
 
     @pytest.mark.asyncio
     async def test_create_run_with_optional_params(self, client, mock_httpx_client, sample_run):
@@ -492,7 +492,7 @@ class TestCreateRun:
                 dataset_id=dataset_id,
                 target_column_id=target_column_id,
                 task="regression",
-                mode="fast",
+                depth_iterations=1,
                 timeseries_groups=timeseries_groups,
                 target_column_override="price_override",
                 author="Test Author",
@@ -680,7 +680,7 @@ class TestRunAsync:
             result = await client.run_async(
                 file=df,
                 target_column="price",
-                mode="fast",
+                depth_iterations=1,
             )
 
         assert isinstance(result, EngineResult)
@@ -795,7 +795,7 @@ class TestRun:
             result = client.run(
                 file=df,
                 target_column="price",
-                mode="fast",
+                depth_iterations=1,
             )
 
         assert isinstance(result, EngineResult)

{discovery_engine_api-0.1.34 → discovery_engine_api-0.1.37}/tests/test_client_e2e.py

@@ -142,7 +142,7 @@ async def test_client_e2e_full_flow(engine, test_dataframe):
     result = await engine.run_async(
         file=test_dataframe,
         target_column="price",
-        mode="fast",
+        depth_iterations=1,
         description="E2E test dataset - house price prediction",
         column_descriptions={
             "age": "Age of the property owner",
@@ -183,7 +183,7 @@ async def test_client_e2e_async_workflow(engine, test_dataframe):
     result = await engine.run_async(
         file=test_dataframe,
         target_column="price",
-        mode="fast",
+        depth_iterations=1,
         auto_report_use_llm_evals=False,
         wait=False,  # Don't wait immediately
     )
@@ -220,7 +220,7 @@ async def test_client_e2e_get_results(engine, test_dataframe):
     result = await engine.run_async(
         file=test_dataframe,
         target_column="price",
-        mode="fast",
+        depth_iterations=1,
         auto_report_use_llm_evals=False,
         wait=False,
     )