Merge pull request #166 from emmanuel-ferdman/main

ClaudioZeni · web-flow · commit 0eded609a02c · 2025-06-10T13:18:39.000+01:00
Modernize logger interface
diff --git a/mattergen/diffusion/corruption/corruption.py b/mattergen/diffusion/corruption/corruption.py
@@ -49,7 +49,7 @@ def maybe_expand(x: torch.Tensor, batch: B, like: torch.Tensor = None) -> torch.
         return x
     else:
         if x.shape[0] == batch.shape[0]:
-            logging.warn(
+            logging.warning(
                 "Warning: batch shape is == x shape, are you trying to expand something that is already expanded?"
             )
         return x[batch]

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ def maybe_expand(x: torch.Tensor, batch: B, like: torch.Tensor = None) -> torch.`
`49`	`49`	`return x`
`50`	`50`	`else:`
`51`	`51`	`if x.shape[0] == batch.shape[0]:`
`52`		`- logging.warn(`
	`52`	`+ logging.warning(`
`53`	`53`	`"Warning: batch shape is == x shape, are you trying to expand something that is already expanded?"`
`54`	`54`	`)`
`55`	`55`	`return x[batch]`