fix(bigquery): fix repr of model reference (#66)

googleapis · Mar 26, 2020 · 26c6204 · 26c6204
1 parent 2abdef8
commit 26c6204
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/google/cloud/bigquery/model.py b/google/cloud/bigquery/model.py
@@ -430,6 +430,6 @@ def __hash__(self):
         return hash(self._key())
 
     def __repr__(self):
-        return "ModelReference(project='{}', dataset_id='{}', project_id='{}')".format(
+        return "ModelReference(project_id='{}', dataset_id='{}', model_id='{}')".format(
             self.project, self.dataset_id, self.model_id
         )
diff --git a/tests/unit/model/test_model.py b/tests/unit/model/test_model.py
@@ -316,5 +316,5 @@ def test_repr(target_class):
     got = repr(model)
     assert got == (
         "Model(reference=ModelReference("
-        "project='my-proj', dataset_id='my_dset', project_id='my_model'))"
+        "project_id='my-proj', dataset_id='my_dset', model_id='my_model'))"
     )
diff --git a/tests/unit/model/test_model_reference.py b/tests/unit/model/test_model_reference.py
@@ -136,5 +136,5 @@ def test_repr(target_class):
     got = repr(model)
     assert (
         got
-        == "ModelReference(project='my-proj', dataset_id='my_dset', project_id='my_model')"
+        == "ModelReference(project_id='my-proj', dataset_id='my_dset', model_id='my_model')"
     )