featrixsphere 0.2.1438__py3-none-any.whl → 0.2.1461__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
featrixsphere/__init__.py CHANGED
@@ -38,7 +38,7 @@ Example:
38
38
  ... labels=['Experiment A', 'Experiment B'])
39
39
  """
40
40
 
41
- __version__ = "0.2.1438"
41
+ __version__ = "0.2.1461"
42
42
  __author__ = "Featrix"
43
43
  __email__ = "support@featrix.com"
44
44
  __license__ = "MIT"
featrixsphere/client.py CHANGED
@@ -1635,7 +1635,7 @@ class FeatrixSphereClient:
1635
1635
  ax.text(0.5, 0.5, f'Error plotting embedding: {e}',
1636
1636
  transform=ax.transAxes, ha='center', va='center')
1637
1637
 
1638
- def create_embedding_space(self, name: str, s3_training_dataset: str, s3_validation_dataset: str, webhooks: Dict[str, str] = None) -> SessionInfo:
1638
+ def create_embedding_space(self, name: str, s3_training_dataset: str, s3_validation_dataset: str, webhooks: Dict[str, str] = None, user_metadata: Dict[str, Any] = None) -> SessionInfo:
1639
1639
  """
1640
1640
  Create a new embedding space from S3 training and validation datasets.
1641
1641
 
@@ -1644,6 +1644,7 @@ class FeatrixSphereClient:
1644
1644
  s3_training_dataset: S3 URL for training dataset (must start with 's3://')
1645
1645
  s3_validation_dataset: S3 URL for validation dataset (must start with 's3://')
1646
1646
  webhooks: Optional dict with webhook configuration keys (webhook_callback_secret, s3_backup_url, model_id_update_url)
1647
+ user_metadata: Optional user metadata for ES/SP identification (max 32KB)
1647
1648
 
1648
1649
  Returns:
1649
1650
  SessionInfo for the newly created embedding space session
@@ -1668,6 +1669,10 @@ class FeatrixSphereClient:
1668
1669
  }
1669
1670
  if webhooks:
1670
1671
  data['webhooks'] = webhooks
1672
+ if user_metadata:
1673
+ import json
1674
+ data['user_metadata'] = json.dumps(user_metadata)
1675
+ print(f"User metadata: {user_metadata}")
1671
1676
 
1672
1677
  response_data = self._post_json("/compute/create-embedding-space", data)
1673
1678
 
@@ -1689,7 +1694,8 @@ class FeatrixSphereClient:
1689
1694
  parent_embedding_space_path: str = None,
1690
1695
  s3_training_dataset: str = None,
1691
1696
  s3_validation_dataset: str = None,
1692
- webhooks: Dict[str, str] = None
1697
+ webhooks: Dict[str, str] = None,
1698
+ user_metadata: Dict[str, Any] = None
1693
1699
  ) -> SessionInfo:
1694
1700
  """
1695
1701
  Fine-tune an existing embedding space on new data.
@@ -1785,6 +1791,10 @@ class FeatrixSphereClient:
1785
1791
 
1786
1792
  if webhooks:
1787
1793
  data['webhooks'] = webhooks
1794
+ if user_metadata:
1795
+ import json
1796
+ data['user_metadata'] = json.dumps(user_metadata)
1797
+ print(f"User metadata: {user_metadata}")
1788
1798
 
1789
1799
  response_data = self._post_json("/compute/fine-tune-embedding-space", data)
1790
1800
 
@@ -4203,7 +4213,8 @@ class FeatrixSphereClient:
4203
4213
  class_imbalance: dict = None,
4204
4214
  poll_interval: int = 30, max_poll_time: int = 3600,
4205
4215
  verbose: bool = True,
4206
- webhooks: Dict[str, str] = None) -> SessionInfo:
4216
+ webhooks: Dict[str, str] = None,
4217
+ user_metadata: Dict[str, Any] = None) -> SessionInfo:
4207
4218
  """
4208
4219
  Train a single predictor on a foundation model (existing embedding space).
4209
4220
 
@@ -4227,6 +4238,7 @@ class FeatrixSphereClient:
4227
4238
  max_poll_time: Maximum time to poll in seconds (default: 3600 = 1 hour)
4228
4239
  verbose: Whether to print status updates during polling (default: True)
4229
4240
  webhooks: Optional dict with webhook configuration keys (webhook_callback_secret, s3_backup_url, model_id_update_url)
4241
+ user_metadata: Optional user metadata for ES/SP identification (max 32KB)
4230
4242
 
4231
4243
  Returns:
4232
4244
  SessionInfo for the newly created predictor training session
@@ -4318,6 +4330,11 @@ class FeatrixSphereClient:
4318
4330
  if webhooks:
4319
4331
  import json
4320
4332
  data["webhooks"] = json.dumps(webhooks)
4333
+ if user_metadata:
4334
+ import json
4335
+ data["user_metadata"] = json.dumps(user_metadata)
4336
+ if verbose:
4337
+ print(f"User metadata: {user_metadata}")
4321
4338
 
4322
4339
  # Send request with file if provided
4323
4340
  try:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: featrixsphere
3
- Version: 0.2.1438
3
+ Version: 0.2.1461
4
4
  Summary: Transform any CSV into a production-ready ML model in minutes, not months.
5
5
  Home-page: https://github.com/Featrix/sphere
6
6
  Author: Featrix
@@ -0,0 +1,9 @@
1
+ featrixsphere/__init__.py,sha256=lES5H__B_mdE-8UdrooDOZ0ttmWZ4VuBt0ub6-cHkxQ,1888
2
+ featrixsphere/cli.py,sha256=AW9O3vCvCNJ2UxVGN66eRmeN7XLSiHJlvK6JLZ9UJXc,13358
3
+ featrixsphere/client.py,sha256=P7tLWX7yfo0w3Ctug71N3Jk124ITJpiqs1dH6wk3iBY,385519
4
+ featrixsphere/test_client.py,sha256=4SiRbib0ms3poK0UpnUv4G0HFQSzidF3Iswo_J2cjLk,11981
5
+ featrixsphere-0.2.1461.dist-info/METADATA,sha256=-df_LYBCPHyIYHS4x39EclimcS7vYcVwYVA3KPSkOcg,16232
6
+ featrixsphere-0.2.1461.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
7
+ featrixsphere-0.2.1461.dist-info/entry_points.txt,sha256=QreJeYfD_VWvbEqPmMXZ3pqqlFlJ1qZb-NtqnyhEldc,51
8
+ featrixsphere-0.2.1461.dist-info/top_level.txt,sha256=AyN4wjfzlD0hWnDieuEHX0KckphIk_aC73XCG4df5uU,14
9
+ featrixsphere-0.2.1461.dist-info/RECORD,,
@@ -1,9 +0,0 @@
1
- featrixsphere/__init__.py,sha256=0gIgPqNrcFK35jlMQqSLQg9TqBwaiNU_PhZ4dQ3JOsw,1888
2
- featrixsphere/cli.py,sha256=AW9O3vCvCNJ2UxVGN66eRmeN7XLSiHJlvK6JLZ9UJXc,13358
3
- featrixsphere/client.py,sha256=NM-dBRANvN0GpcLf0BPqUxpqwMhzHP6hJktD8FndyCM,384646
4
- featrixsphere/test_client.py,sha256=4SiRbib0ms3poK0UpnUv4G0HFQSzidF3Iswo_J2cjLk,11981
5
- featrixsphere-0.2.1438.dist-info/METADATA,sha256=tNsudbOj5N1zFcjNA2iHZf6WJa4jxz1_MfhctYs8-qI,16232
6
- featrixsphere-0.2.1438.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
7
- featrixsphere-0.2.1438.dist-info/entry_points.txt,sha256=QreJeYfD_VWvbEqPmMXZ3pqqlFlJ1qZb-NtqnyhEldc,51
8
- featrixsphere-0.2.1438.dist-info/top_level.txt,sha256=AyN4wjfzlD0hWnDieuEHX0KckphIk_aC73XCG4df5uU,14
9
- featrixsphere-0.2.1438.dist-info/RECORD,,