Add components sdks to client
@@ -36,6 +36,8 @@ class PolyaxonClient:
self._users_v1 = None
self._versions_v1 = None
self._agents_v1 = None
+ self._components_v1 = None
+ self._models_v1 = None
def reset(self):
self._transport = None
@@ -45,6 +47,8 @@ class PolyaxonClient:
self._users_v1 = None
self._versions_v1 = None
self._agents_v1 = None
+ self._components_v1 = None
+ self._models_v1 = None
self.api_client = polyaxon_sdk.ApiClient(
self.config.sdk_config, **self.config.client_header
)
@@ -101,5 +105,17 @@ class PolyaxonClient:
self._agents_v1 = polyaxon_sdk.AgentsV1Api(self.api_client)
return self._agents_v1
+ @property
+ def components_v1(self):
+ if not self._components_v1:
+ self._components_v1 = polyaxon_sdk.HubComponentsV1Api(self.api_client)
+ return self._components_v1
+
+ @property
+ def models_v1(self):
+ if not self._models_v1:
+ self._models_v1 = polyaxon_sdk.HubModelsV1Api(self.api_client)
+ return self._models_v1
+
def sanitize_for_serialization(self, value):
return self.api_client.sanitize_for_serialization(value)
