Skip to content

Commit 34c2980

Browse files
authored
Convert models (#647)
This finishes the conversion of the API methods. Reminder: `kagglesdk` is generated by `kapigen` and can be ignored. Also, `kaggle` and `src/kaggle` have the same changes so only one needs to be reviewed.
1 parent bb6867d commit 34c2980

File tree

12 files changed

+921
-665
lines changed

12 files changed

+921
-665
lines changed

integration_tests/test_models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
MODEL_HANDLE = "keras/bert"
88
MODEL_ID = 2819
99

10+
# TODO(messick) Add a test that creates a dataset w/o specifying privacy that is created private.
1011

1112
class TestModels(unittest.TestCase):
1213
def setUp(self):
@@ -19,4 +20,4 @@ def test_list_models(self) -> None:
1920

2021
def test_get_model(self) -> None:
2122
model = self.api.model_get(MODEL_HANDLE)
22-
self.assertEqual(MODEL_ID, model['id'])
23+
self.assertEqual(MODEL_ID, model.id)

0 commit comments

Comments
 (0)