Skip to content

class name change for documentation #30

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Timo-SH merged 2 commits into from
Jul 6, 2026
Merged

class name change for documentation #30

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ec7374b
class name change for documentation
Timo-SH Jul 6, 2026
9717ece
class name change for documentation
Timo-SH Jul 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions graphbench/_loader/_loader.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from graphbench._metadata import expand_dataset_names
from ._dataset_registry import DatasetRegistry
from ._split_strategies import AlgoReasSplitStrategy, FixedSplitStrategy, RatioSplitStrategy, TrainValTestSet
from graphbench.datasets._weatherforecasting import _prepare_weather_cache_once, EfficientWeatherGraphDataset
from graphbench.datasets._weatherforecasting import _prepare_weather_cache_once, WeatherforecastingDataset


class Loader():
Expand Down Expand Up @@ -263,7 +263,7 @@ def _make_weather_dataset(
#dataset = EfficientWeatherGraphDataset(root=root, pre_transform=None, transform=None)

#hier noch den return anpassen
return EfficientWeatherGraphDataset(
return WeatherforecastingDataset(
root=self.root,
pre_filter=self.pre_filter,
pre_transform=self.pre_transform,
Expand Down
4 changes: 2 additions & 2 deletions graphbench/datasets/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from ._combinatorial_optimization import CODataset
from ._electroniccircuits import ECDataset
from ._sat import SATDataset
from ._weatherforecasting import EfficientWeatherGraphDataset
from ._weatherforecasting import WeatherforecastingDataset


__all__ = [
Expand All @@ -16,5 +16,5 @@
"CODataset",
"ECDataset",
"SATDataset",
"EfficientWeatherGraphDataset",
"WeatherforecastingDataset",
]
8 changes: 4 additions & 4 deletions graphbench/datasets/_weatherforecasting.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def tqdm(iterable, *args, **kwargs):

# Data files specific to each task (sourced from the task-specific repo).
# weather_64.pt is the raw timestep data; saved as weather_dataset.pt which is
# what EfficientWeatherGraphDataset.process() expects.
# what WeatherforecastingDataset.process() expects.
_HF_WEATHER_DATA_FILES = {
"weather_64.pt": "weather_dataset.pt",
}
Expand Down Expand Up @@ -114,7 +114,7 @@ def _prepare_weather_cache_once(root: str, timeout_s: int, skip_cache_build: boo
Raw component files (static_components.pkl, metadata.pkl, etc.) are
downloaded from HuggingFace automatically when missing.
"""
#from wf_utils import EfficientWeatherGraphDataset
#from wf_utils import WeatherforecastingDataset

root_path = Path(root)
processed_path = root_path /"weather" / "processed" / "weather_graph_data_processed.pt"
Expand All @@ -137,7 +137,7 @@ def _prepare_weather_cache_once(root: str, timeout_s: int, skip_cache_build: boo
elif not processed_path.exists():
_ensure_raw_weather_files(root, task_name)
logger.info(f"Rank 0 building weather processed cache at {processed_path}")
EfficientWeatherGraphDataset(root=root, pre_transform=None, transform=None)
WeatherforecastingDataset(root=root, pre_transform=None, transform=None)
else:
logger.info(f"Rank 0 reusing existing weather processed cache at {processed_path}")

Expand Down Expand Up @@ -412,7 +412,7 @@ def backmap(sorted_idx: np.ndarray) -> np.ndarray:

return TemporalSplits(train_idx=train_idx, val_idx=val_idx, test_idx=test_idx)

class EfficientWeatherGraphDataset(InMemoryDataset):
class WeatherforecastingDataset(InMemoryDataset):
"""
Weather forecasting dataset.

Expand Down
Loading