Skip to content

Commit c0023aa

Browse files
nits fixes
1 parent 46f420d commit c0023aa

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

nucleus/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def list_models(self) -> List[Model]:
139139
model_objects = self._make_request({}, "models/", requests.get)
140140

141141
try:
142-
models = [Model(model['id'], model["name"], model["ref_id"], model["metadata"], self) for model in model_objects['models']]
142+
models = [Model(model["id"], model["name"], model["ref_id"], model["metadata"], self) for model in model_objects["models"]]
143143
return models
144144
except e:
145145
return []

nucleus/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def __init__(
2525
self._client = client
2626

2727
def __repr__(self):
28-
return f'Model({self.id}, {self.name}, {self.reference_id}, {self.metadata}, {self._client})'
28+
return f'Model(model_id={self.id}, name={self.name}, reference_id={self.reference_id}, metadata={self.metadata}, client={self._client})'
2929

3030
def create_run(
3131
self,

0 commit comments

Comments
 (0)