update test w/ new cluster_studio_dashboard arguments
@@ -58,7 +58,7 @@ def upload_data(db_name):
)
-@pytest.mark.skip(reason="AWS Connection Required")
+# @pytest.mark.skip(reason="AWS Connection Required")
def test_full_example_athena(tmp_path):
session_read, db_name_read = setup_athena_db()
@@ -99,8 +99,8 @@ def test_full_example_athena(tmp_path):
linker.cluster_studio_dashboard(
df_predict,
df_clusters,
- [0, 4],
- os.path.join(tmp_path, "test_cluster_studio.html"),
+ sampling_method="by_cluster_size",
+ out_path=os.path.join(tmp_path, "test_cluster_studio.html"),
)
linker.unlinkables_chart(source_dataset="Testing")
