From f9938734db8f17cc372ce5d6051ba905ac448500 Mon Sep 17 00:00:00 2001 From: Tiziano Date: Thu, 29 Jan 2026 14:21:15 +0100 Subject: [PATCH] refactor: remove all main functions --- hyperbench/data/dataset.py | 12 ------------ hyperbench/tests/train/trainer_test.py | 4 ---- hyperbench/tests/utils/data_utils_test.py | 4 ---- 3 files changed, 20 deletions(-) diff --git a/hyperbench/data/dataset.py b/hyperbench/data/dataset.py index e2a2172..70c3a1d 100644 --- a/hyperbench/data/dataset.py +++ b/hyperbench/data/dataset.py @@ -417,15 +417,3 @@ class DBLPDataset(Dataset): class ThreadsMathsxDataset(Dataset): DATASET_NAME = "THREADSMATHSX" - - -if __name__ == "__main__": - for dataset in DatasetNames: - print(f"Processing dataset: {dataset.value}") - if dataset == DatasetNames.EMAIL_ENRON: - load_hif = HIFConverter.load_from_hif(dataset.name, save_on_disk=True) - continue - load_hif = HIFConverter.load_from_hif(dataset.name) - print( - f"Loaded HIF hypergraph with {len(load_hif.nodes)} nodes and {len(load_hif.edges)} edges." - ) diff --git a/hyperbench/tests/train/trainer_test.py b/hyperbench/tests/train/trainer_test.py index 64ae672..c82f121 100644 --- a/hyperbench/tests/train/trainer_test.py +++ b/hyperbench/tests/train/trainer_test.py @@ -217,7 +217,3 @@ def test_test_all_with_verbose_true_prints(_, mock_model_configs, caplog): record.message for record in caplog.records if "Test model" in record.message ] assert len(logs) == len(mock_model_configs) - - -if __name__ == "__main__": - pytest.main([__file__, "-v"]) diff --git a/hyperbench/tests/utils/data_utils_test.py b/hyperbench/tests/utils/data_utils_test.py index 9ba11a1..d0f18bb 100644 --- a/hyperbench/tests/utils/data_utils_test.py +++ b/hyperbench/tests/utils/data_utils_test.py @@ -97,7 +97,3 @@ def test_empty_hifhypergraph(): assert result.edges == [] assert result.incidences == [] assert result.metadata == {} - - -if __name__ == "__main__": - pytest.main([__file__, "-v"])