Fixing tests

bigmlcom · Nov 26, 2022 · 560852d · 560852d
1 parent f42c64f
commit 560852d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/bigml/tests/create_cluster_steps.py b/bigml/tests/create_cluster_steps.py
@@ -43,7 +43,7 @@ def i_create_a_cluster(step, shared=None):
 
 def i_create_a_cluster_from_dataset_list(step):
     """Step: I create a cluster from a dataset list"""
-    resource = world.api.create_cluster(world.dataset_ids)
+    resource = world.api.create_cluster(step.bigml["dataset_ids"])
     world.status = resource['code']
     eq_(world.status, HTTP_CREATED)
     world.location = resource['location']

diff --git a/bigml/tests/create_lda_steps.py b/bigml/tests/create_lda_steps.py
@@ -43,7 +43,7 @@ def i_create_a_topic_model(step):
 
 def i_create_a_topic_model_from_dataset_list(step):
     """Step: I create a topic model from a dataset list"""
-    resource = world.api.create_topic_model(world.dataset_ids)
+    resource = world.api.create_topic_model(step.bigml["dataset_ids"])
     world.status = resource['code']
     eq_(world.status, HTTP_CREATED)
     world.location = resource['location']

diff --git a/bigml/tests/create_model_steps.py b/bigml/tests/create_model_steps.py
@@ -79,7 +79,7 @@ def i_create_a_balanced_model(step):
 
 def i_create_a_model_from_dataset_list(step):
     """Step: I create a model from a dataset list"""
-    resource = world.api.create_model(world.dataset_ids,
+    resource = world.api.create_model(step.bigml["dataset_ids"],
                                       args=NO_MISSING_SPLITS)
     world.status = resource['code']
     eq_(world.status, HTTP_CREATED)