Skip to content

Commit f966c1b

Browse files
committed
refactor: update AtomsDataModule deprecation warning and configuration path
1 parent 1969fbf commit f966c1b

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/schnetpack/configs/data/custom.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# @package data
2-
_target_: schnetpack.data.datamodule_v2.AtomsDataModuleV2
2+
_target_: schnetpack.data.datamodule.AtomsDataModule
33

44
dataset:
55
_target_: schnetpack.data.ASEAtomsData

src/schnetpack/data/datamodule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def __init__(
7878

7979
if used_legacy_args:
8080
warnings.warn(
81-
"The following arguments are deprecated in `AtomsDataModuleV2`: "
81+
"The following arguments are deprecated in `AtomsDataModule`: "
8282
f"{used_legacy_args}. "
8383
"Use a prebuilt dataset instance and configure these options on the "
8484
"dataset instead.",

0 commit comments

Comments
 (0)