Skip to content

Commit 19f4128

Browse files
Merge pull request #203 from Paperspace/PS-12681-Fix_arguments_of_ExperimentsClient.list
Fix ExperimentClient.list method parameters
2 parents c25100f + ad2d545 commit 19f4128

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

gradient/api_sdk/clients/experiment_client.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -706,16 +706,20 @@ def stop(self, experiment_id, use_vpc=False):
706706
repository = repositories.StopExperiment(api_key=self.api_key, logger=self.logger)
707707
repository.stop(experiment_id, use_vpc=use_vpc)
708708

709-
def list(self, project_id=None, **kwargs):
709+
def list(self, project_id=None, offset=None, limit=None, get_meta=False):
710710
"""Get a list of experiments. Optionally filter by project ID
711711
712712
:param str|list|None project_id:
713+
:param int offset:
714+
:param int limit:
715+
:param bool get_meta: get dict of metadata like number of total items, etc. Setting to True changes rtype
716+
713717
:return: experiments
714-
:rtype: list[models.SingleNodeExperiment|models.MultiNodeExperiment]
718+
:rtype: list[models.SingleNodeExperiment|models.MultiNodeExperiment]|tuple[list[models.SingleNodeExperiment|models.MultiNodeExperiment],dict]
715719
"""
716720

717721
repository = repositories.ListExperiments(api_key=self.api_key, logger=self.logger)
718-
experiments = repository.list(project_id=project_id, **kwargs)
722+
experiments = repository.list(project_id=project_id, limit=limit, offset=offset, get_meta=get_meta)
719723
return experiments
720724

721725
def get(self, experiment_id):

0 commit comments

Comments
 (0)