sutro 0.1.22__tar.gz → 0.1.24__tar.gz

{sutro-0.1.22 → sutro-0.1.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sutro
-Version: 0.1.22
+Version: 0.1.24
 Summary: Sutro Python SDK
 Project-URL: Homepage, https://sutro.sh
 Project-URL: Documentation, https://docs.sutro.sh

{sutro-0.1.22 → sutro-0.1.24}/pyproject.toml

@@ -9,7 +9,7 @@ installer = "uv"
 
 [project]
 name = "sutro"
-version = "0.1.22"
+version = "0.1.24"
 description = "Sutro Python SDK"
 readme = "README.md"
 requires-python = ">=3.10"

{sutro-0.1.22 → sutro-0.1.24}/sutro/sdk.py

@@ -410,7 +410,7 @@ class Sutro:
                     spinner.stop()
                     return None
 
-                results = job_results_response.json()["results"]
+                results = job_results_response.json()["results"]["outputs"]
 
                 spinner.write(
                     to_colored_text(
@@ -421,12 +421,6 @@ class Sutro:
                 spinner.stop()
 
                 if isinstance(data, (pd.DataFrame, pl.DataFrame)):
-                    sample_n = 1 if sampling_params is None else sampling_params["n"]
-                    if sample_n > 1:
-                        results = [
-                            results[i : i + sample_n]
-                            for i in range(0, len(results), sample_n)
-                        ]
                     if isinstance(data, pd.DataFrame):
                         data[output_column] = results
                     elif isinstance(data, pl.DataFrame):
@@ -875,15 +869,7 @@ class Sutro:
         response_data = response.json()
         results_df = pl.DataFrame(response_data["results"])
 
-
-        if len(results_df.columns ) == 1:
-            # Default column when API is only returning a list, and we construct the df
-            # from that
-            original_results_column = 'column_0'
-        else:
-            original_results_column = 'outputs'
-
-        results_df = results_df.rename({original_results_column: output_column})
+        results_df = results_df.rename({'outputs': output_column})
 
         # Ordering inputs col first seems most logical/useful
         column_config = [