Skip to content

Commit cce3699

Browse files
authored
remove extra assert; blocks edge_chunk_size!=None (#1633)
1 parent ac1a4dd commit cce3699

1 file changed

Lines changed: 0 additions & 6 deletions

File tree

src/fairchem/core/units/mlip_unit/predict.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -250,12 +250,6 @@ def predict(
250250
self.model = torch.compile(self.model, dynamic=True)
251251
self.lazy_model_intialized = True
252252

253-
if self.inference_mode.external_graph_gen and data.edge_index.shape[1] == 0:
254-
raise ValueError(
255-
"Cannot run inference with external graph generation on empty edge index. "
256-
"Please ensure the input data has valid edges."
257-
)
258-
259253
# this needs to be .clone() to avoid issues with graph parallel modifying this data with MOLE
260254
data_device = data.to(self.device).clone()
261255

0 commit comments

Comments
 (0)