synthetic-graph-benchmarks 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

synthetic_graph_benchmarks/__init__.py

@@ -3,8 +3,9 @@ from synthetic_graph_benchmarks.benchmarks import (
     benchmark_planar_results,
     benchmark_tree_results,
 )
+from synthetic_graph_benchmarks.dataset import Dataset
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 def main():
     """Main entry point for the CLI."""
@@ -20,4 +21,5 @@ __all__ = [
     "benchmark_planar_results", 
     "benchmark_tree_results",
     "__version__",
+    "Dataset",
 ]

synthetic_graph_benchmarks/dataset.py

@@ -16,7 +16,7 @@ class Dataset:
     
     
     @classmethod
-    def load_from_pickle_url(cls, url: str):
+    def load_from_pickle_url(cls, url: str, cache_dir: str = "data") -> "Dataset":
         """
         Load a dataset from a pickle file available at the given URL.
         
@@ -26,7 +26,7 @@ class Dataset:
         Returns:
             Dataset: An instance of the Dataset class with loaded graphs.
         """
-        res = download_file(url, "data")
+        res = download_file(url, cache_dir)
         with open(res, "rb") as f:
             data = pickle.load(f)
         return cls(
@@ -35,13 +35,16 @@ class Dataset:
             test_graphs=data.get('test', None)
         )
     @classmethod
-    def load_sbm(cls):
-        return cls.load_from_pickle_url("https://raw.githubusercontent.com/AndreasBergmeister/graph-generation/main/data/sbm.pkl")
+    def load_sbm(cls, cache_dir: str = "data") -> "Dataset":
+        """Load a dataset of Stochastic Block Model (SBM) graphs."""
+        return cls.load_from_pickle_url("https://raw.githubusercontent.com/AndreasBergmeister/graph-generation/main/data/sbm.pkl", cache_dir)
     
     @classmethod
-    def load_planar(cls):
-        return cls.load_from_pickle_url("https://raw.githubusercontent.com/AndreasBergmeister/graph-generation/main/data/planar.pkl")
-    
+    def load_planar(cls, cache_dir: str = "data") -> "Dataset":
+        """Load a dataset of planar graphs."""
+        return cls.load_from_pickle_url("https://raw.githubusercontent.com/AndreasBergmeister/graph-generation/main/data/planar.pkl", cache_dir)
+
     @classmethod
-    def load_tree(cls):
-        return cls.load_from_pickle_url("https://raw.githubusercontent.com/AndreasBergmeister/graph-generation/main/data/tree.pkl")
+    def load_tree(cls, cache_dir: str = "data") -> "Dataset":
+        """Load a dataset of tree graphs."""
+        return cls.load_from_pickle_url("https://raw.githubusercontent.com/AndreasBergmeister/graph-generation/main/data/tree.pkl", cache_dir)

{synthetic_graph_benchmarks-0.1.0.dist-info → synthetic_graph_benchmarks-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synthetic-graph-benchmarks
-Version: 0.1.0
+Version: 0.1.1
 Summary: Standardized benchmarks for evaluating synthetic graph generation methods
 Project-URL: Homepage, https://github.com/peteole/synthetic_graph_benchmarks
 Project-URL: Repository, https://github.com/peteole/synthetic_graph_benchmarks

synthetic_graph_benchmarks-0.1.1.dist-info/RECORD

@@ -0,0 +1,11 @@
+synthetic_graph_benchmarks/__init__.py,sha256=dyuP3-iM0GGHW4Ufa3dR96z-Mzpn1g-XNhvHoUOIg3I,730
+synthetic_graph_benchmarks/benchmarks.py,sha256=jWDaVrAwe4uYaKl1EnUIVqavDz5ZeuxcSOcHyMwUPBc,2707
+synthetic_graph_benchmarks/dataset.py,sha256=7aJ9kU_Z0VF78-U7rm0uE2wAMG6wEq4KTSiMiUh-SIM,1860
+synthetic_graph_benchmarks/dist_helper.py,sha256=rwtpP_IaIe8DcS5OvrSkYdF8xl7VBHaGoM3jv_Yeeos,7001
+synthetic_graph_benchmarks/spectre_utils.py,sha256=CuOQdTlznar9_FIM66qoYk5zg1OEfa4M8be4WwHsFKA,41734
+synthetic_graph_benchmarks/utils.py,sha256=Y5QTRmGjr79p1Y6h54c6k9kXz-jDveoj0ZjpAMKml50,1835
+synthetic_graph_benchmarks-0.1.1.dist-info/METADATA,sha256=xC1mrsHFNCnCOoG3ZZDZHTmqwQ_3O16rFYT83rGhkkk,8484
+synthetic_graph_benchmarks-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+synthetic_graph_benchmarks-0.1.1.dist-info/entry_points.txt,sha256=KvCdlYfIhAw2srO-7H9XG0jTaAIENxGXprw0B4aClco,79
+synthetic_graph_benchmarks-0.1.1.dist-info/licenses/LICENSE,sha256=VDqzZZ7UjLMPLrsbgpHsMCW-H4msljYwm9z61rQSIsc,1069
+synthetic_graph_benchmarks-0.1.1.dist-info/RECORD,,

synthetic_graph_benchmarks-0.1.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-synthetic_graph_benchmarks/__init__.py,sha256=pMEdpnJXD8laVjCG4zd8d555chyqXv3-2WGxnrKFAQg,660
-synthetic_graph_benchmarks/benchmarks.py,sha256=jWDaVrAwe4uYaKl1EnUIVqavDz5ZeuxcSOcHyMwUPBc,2707
-synthetic_graph_benchmarks/dataset.py,sha256=Yca2fNPvRu7cOFyjjixkyqttboC07iPDgUXZWvvweN0,1515
-synthetic_graph_benchmarks/dist_helper.py,sha256=rwtpP_IaIe8DcS5OvrSkYdF8xl7VBHaGoM3jv_Yeeos,7001
-synthetic_graph_benchmarks/spectre_utils.py,sha256=CuOQdTlznar9_FIM66qoYk5zg1OEfa4M8be4WwHsFKA,41734
-synthetic_graph_benchmarks/utils.py,sha256=Y5QTRmGjr79p1Y6h54c6k9kXz-jDveoj0ZjpAMKml50,1835
-synthetic_graph_benchmarks-0.1.0.dist-info/METADATA,sha256=7zz-Yd3WXLVNjax5UPbkmPWO0n2uMRTD9w531tuqc3I,8484
-synthetic_graph_benchmarks-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-synthetic_graph_benchmarks-0.1.0.dist-info/entry_points.txt,sha256=KvCdlYfIhAw2srO-7H9XG0jTaAIENxGXprw0B4aClco,79
-synthetic_graph_benchmarks-0.1.0.dist-info/licenses/LICENSE,sha256=VDqzZZ7UjLMPLrsbgpHsMCW-H4msljYwm9z61rQSIsc,1069
-synthetic_graph_benchmarks-0.1.0.dist-info/RECORD,,