File tree Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -169,6 +169,11 @@ def n_atoms_per_batch(self) -> torch.Tensor:
169
169
deprecated::
170
170
Use :attr:`n_atoms_per_system` instead.
171
171
"""
172
+ warnings .warn (
173
+ "n_atoms_per_batch is deprecated, use n_atoms_per_system instead" ,
174
+ DeprecationWarning ,
175
+ stacklevel = 2 ,
176
+ )
172
177
return self .n_atoms_per_system
173
178
174
179
@property
@@ -178,6 +183,11 @@ def batch(self) -> torch.Tensor | None:
178
183
deprecated::
179
184
Use :attr:`system_idx` instead.
180
185
"""
186
+ warnings .warn (
187
+ "batch is deprecated, use system_idx instead" ,
188
+ DeprecationWarning ,
189
+ stacklevel = 2 ,
190
+ )
181
191
return self .system_idx
182
192
183
193
@batch .setter
@@ -187,6 +197,11 @@ def batch(self, system_idx: torch.Tensor) -> None:
187
197
deprecated::
188
198
Use :attr:`system_idx` instead.
189
199
"""
200
+ warnings .warn (
201
+ "Setting batch is deprecated, use system_idx instead" ,
202
+ DeprecationWarning ,
203
+ stacklevel = 2 ,
204
+ )
190
205
self .system_idx = system_idx
191
206
192
207
@property
@@ -196,6 +211,11 @@ def n_batches(self) -> int:
196
211
deprecated::
197
212
Use :attr:`n_systems` instead.
198
213
"""
214
+ warnings .warn (
215
+ "n_batches is deprecated, use n_systems instead" ,
216
+ DeprecationWarning ,
217
+ stacklevel = 2 ,
218
+ )
199
219
return self .n_systems
200
220
201
221
@property
You can’t perform that action at this time.
0 commit comments