Disable flaky twitter bots dataset loading test. (#3439)

ludwig-ai · Aug 11, 2023 · 7ecdfa5 · 7ecdfa5
1 parent 3691dbe
commit 7ecdfa5
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/tests/ludwig/datasets/test_dataset_configs.py b/tests/ludwig/datasets/test_dataset_configs.py
@@ -15,15 +15,16 @@ def test_get_config_and_load(tmpdir):
     assert df is not None
     assert len(df) == 18721  # Expected number of rows in Yosemite temperatures dataset.
 
-    # Test loading dataset without 'split' and 'Unnamed: 0' columns in config.
-    twitter_bots_config = ludwig.datasets._get_dataset_config("twitter_bots")
-    assert isinstance(twitter_bots_config, DatasetConfig)
-
-    twitter_bots_dataset = ludwig.datasets.get_dataset("twitter_bots", cache_dir=tmpdir)
-    assert isinstance(twitter_bots_dataset, DatasetLoader)
-    df = twitter_bots_dataset.load()
-    assert df is not None
-    assert len(df.columns) == 22  # Expected number of columns in Twitter bots dataset including split column.
+    # DISABLED: Flaky for tests, probably due to the dataset size.
+    # # Test loading dataset without 'split' and 'Unnamed: 0' columns in config.
+    # twitter_bots_config = ludwig.datasets._get_dataset_config("twitter_bots")
+    # assert isinstance(twitter_bots_config, DatasetConfig)
+
+    # twitter_bots_dataset = ludwig.datasets.get_dataset("twitter_bots", cache_dir=tmpdir)
+    # assert isinstance(twitter_bots_dataset, DatasetLoader)
+    # df = twitter_bots_dataset.load()
+    # assert df is not None
+    # assert len(df.columns) == 22  # Expected number of columns in Twitter bots dataset including split column.
 
 
 def test_get_config_kaggle(tmpdir):