def get_versioning_api() -> VersioningApi:
configuration = Configuration()
configuration.host = getenv('LIGHTLY_SERVER_LOCATION',
'https://api.lightly.ai')
api_client = ApiClient(configuration=configuration)
versioning_api = VersioningApi(api_client)
return versioning_api
def __init__(self, token: str, dataset_id: str = None, embedding_id: str = None):
self.check_version_compatibility()
configuration = Configuration()
configuration.host = getenv('LIGHTLY_SERVER_LOCATION', 'https://api.lightly.ai')
configuration.api_key = {'token': token}
api_client = ApiClient(configuration=configuration)
self.api_client = api_client
self.token = token
if dataset_id is not None:
self._dataset_id = dataset_id
if embedding_id is not None:
self.embedding_id = embedding_id
self.datasets_api = DatasetsApi(api_client=self.api_client)
self.samplings_api = SamplingsApi(api_client=self.api_client)
self.jobs_api = JobsApi(api_client=self.api_client)
self.tags_api = TagsApi(api_client=self.api_client)
self.embeddings_api = EmbeddingsApi(api_client=api_client)
self.mappings_api = MappingsApi(api_client=api_client)
self.scores_api = ScoresApi(api_client=api_client)
self.samples_api = SamplesApi(api_client=api_client)
self.quota_api = QuotaApi(api_client=api_client)
def __init__(self, api_client=None):
if api_client is None:
api_client = ApiClient()
self.api_client = api_client