fix: dml.log_metric() resulting in exception when called from within model code when torch.compile()'ing

Author: sehoffmann

dmlcloud/core/metrics.py

@@ -146,6 +146,7 @@ def add_metric(self, name: str, metric: torchmetrics.Metric):
 
         self.metrics[name] = metric
 
+    @torch.compiler.disable
     def log(self, name: str, value: Any, reduction: str = 'mean', **kwargs):
         if reduction not in ['mean', 'sum', 'min', 'max', 'cat']:
             raise ValueError(f'Invalid reduction {reduction}. Must be one of mean, sum, min, max, cat')