proxyml 0.2.1__tar.gz → 0.2.2__tar.gz

{proxyml-0.2.1 → proxyml-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyml
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for calling the ProxyML API
 Author-email: ProxyML <contact@proxyml.ai>
 License:                                  Apache License

{proxyml-0.2.1 → proxyml-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "proxyml"
-version = "0.2.1"
+version = "0.2.2"
 description = "Python SDK for calling the ProxyML API"
 readme = "README.md"
 license = {file = "LICENSE"}

{proxyml-0.2.1 → proxyml-0.2.2}/src/proxyml/client.py

@@ -156,13 +156,13 @@ def delete(endpoint: str) -> requests.models.Response | _ErrorResponse:
         return _ErrorResponse()
 
 
-def put_schema(schema: dict, name: str = "default") -> Any:
+def put_schema(schema: dict, name: str) -> Any:
     """
     Uploads a data schema.
 
     Args:
         schema (dict): data schema object
-        name (str): optional name for the schema, defaults to "default".
+        name (str): name for the schema.
     Returns:
         API response JSON object, or None if the return code was not 200.
     """
@@ -178,7 +178,7 @@ def put_schema(schema: dict, name: str = "default") -> Any:
     return None
 
 
-def fetch_schema(name: str = "default") -> dict | None:
+def fetch_schema(name: str) -> dict | None:
     """Retrieve a stored feature schema by name."""
     r = get(endpoint=f'/schema/{name}', params={})
     if r.status_code == 200:
@@ -242,7 +242,7 @@ def _cast_column(series: pd.Series, ftype: str) -> pd.Series:
     return series
 
 
-def synthesize_data(num_points: int = 100, sample: list | None = None, as_df: bool = True, schema_name: str = "default") -> Any:
+def synthesize_data(num_points: int = 100, sample: list | None = None, as_df: bool = True, *, schema_name: str) -> Any:
     """
     Synthesizes data based on a data schema.
 
@@ -250,7 +250,7 @@ def synthesize_data(num_points: int = 100, sample: list | None = None, as_df: bo
         num_points (int): number of samples to synthesize, defaults to 100.
         sample (list): if specified, synthesized data will be a blend of samples generated from the schema and a number of perturbations around this sample. If not specified (default), all synthesized data is generated by sampling from the data schema.
         as_df (bool): if True (default), data are returned as a pandas DataFrame.
-        schema_name (str): name of the data schema to use to generate the data, defaults to "default."
+        schema_name (str): name of the data schema to use to generate the data.
     Returns:
         JSON object representing the synthesized data if as_df=False, a pandas DataFrame if as_df=True, or None if an error occurred.
     """
@@ -280,7 +280,8 @@ def train_surrogate(
         feature_names: list[str] | None,
         task: str = 'auto',
         test_size: float = 0.2,
-        schema_name: str = "default",
+        *,
+        schema_name: str,
         name: str | None = None,
         comments: str | None = None,
     ) -> Any:
@@ -293,7 +294,7 @@ def train_surrogate(
         feature_names (list): names of the features (columns) in the data.
         task (str): specifies the modeling task, one of "classification," "regression," or "auto" in which case ProxyML will attempt to automatically determine the modeling task.
         test_size (float): fraction of the data to set aside for test data, defaults to 0.2.
-        schema_name (str): name of the data schema to use, defaults to "default."
+        schema_name (str): name of the data schema to use.
         name (str): an optional name for the surrogate.
         comments (str): an optional comment string for the surrogate.
     Returns:

{proxyml-0.2.1 → proxyml-0.2.2}/src/proxyml.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyml
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for calling the ProxyML API
 Author-email: ProxyML <contact@proxyml.ai>
 License:                                  Apache License

{proxyml-0.2.1 → proxyml-0.2.2}/tests/test_client.py

@@ -201,11 +201,11 @@ def _mock_response(status_code, json_body):
 
 
 @patch("proxyml.client.put")
-def test_put_schema_default(mock_put):
+def test_put_schema_success(mock_put):
     mock_put.return_value = _mock_response(200, {"features": []})
-    result = put_schema({"features": []})
+    result = put_schema({"features": []}, name="myschema")
     assert result == {"features": []}
-    mock_put.assert_called_once_with(endpoint="/schema/default", payload={"features": []})
+    mock_put.assert_called_once_with(endpoint="/schema/myschema", payload={"features": []})
 
 
 @patch("proxyml.client.put")
@@ -218,7 +218,7 @@ def test_put_schema_named(mock_put):
 @patch("proxyml.client.put")
 def test_put_schema_failure(mock_put):
     mock_put.return_value = _mock_response(422, {"detail": "invalid"})
-    result = put_schema({"features": []})
+    result = put_schema({"features": []}, name="myschema")
     assert result is None
 
 
@@ -227,11 +227,11 @@ def test_put_schema_failure(mock_put):
 # ---------------------------------------------------------------------------
 
 @patch("proxyml.client.get")
-def test_fetch_schema_default(mock_get):
+def test_fetch_schema_success(mock_get):
     mock_get.return_value = _mock_response(200, {"features": [{"type": "continuous", "name": "age"}]})
-    result = fetch_schema()
+    result = fetch_schema(name="myschema")
     assert result["features"][0]["name"] == "age"
-    mock_get.assert_called_once_with(endpoint="/schema/default", params={})
+    mock_get.assert_called_once_with(endpoint="/schema/myschema", params={})
 
 
 @patch("proxyml.client.get")
@@ -309,12 +309,12 @@ def test_synthesize_data_no_sample(mock_post):
         "feature_names": ["f_cont", "f_cat"],
         "feature_types": ["continuous", "categorical"],
     })
-    df = synthesize_data(num_points=2, sample=None)
+    df = synthesize_data(num_points=2, sample=None, schema_name="myschema")
     assert isinstance(df, pd.DataFrame)
     assert list(df.columns) == ["f_cont", "f_cat"]
     payload = mock_post.call_args.kwargs["payload"]
     assert payload["n"] == 2
-    assert payload["schema_name"] == "default"
+    assert payload["schema_name"] == "myschema"
 
 
 @patch("proxyml.client.post")
@@ -330,7 +330,7 @@ def test_synthesize_data_named_schema(mock_post):
 @patch("proxyml.client.post")
 def test_synthesize_data_failure_returns_none(mock_post):
     mock_post.return_value = _mock_response(500, {})
-    result = synthesize_data(num_points=10, sample=None)
+    result = synthesize_data(num_points=10, sample=None, schema_name="myschema")
     assert result is None
 
 
@@ -343,12 +343,12 @@ def test_train_surrogate_with_metadata(mock_post):
     })
     result = train_surrogate(
         samples=[[1.0, 2.0]], predictions=[3.0],
-        feature_names=None, name="v1", comments="test run",
+        feature_names=None, schema_name="myschema", name="v1", comments="test run",
     )
     payload = mock_post.call_args.kwargs["payload"]
     assert payload["name"] == "v1"
     assert payload["comments"] == "test run"
-    assert payload["schema_name"] == "default"
+    assert payload["schema_name"] == "myschema"
     assert result["version"] == "abc-123"
     assert result["trained_at"] == "2026-04-19T12:00:00"
 
@@ -364,7 +364,7 @@ def test_train_surrogate_named_schema(mock_post):
 @patch("proxyml.client.post")
 def test_train_surrogate_omits_none_metadata(mock_post):
     mock_post.return_value = _mock_response(200, {"version": "abc-123"})
-    train_surrogate(samples=[[1.0]], predictions=[1.0], feature_names=None)
+    train_surrogate(samples=[[1.0]], predictions=[1.0], feature_names=None, schema_name="myschema")
     payload = mock_post.call_args.kwargs["payload"]
     assert "name" not in payload
     assert "comments" not in payload