Logging consistency fixes

Author: thetianshuhuang

grt/evaluate.py

@@ -205,8 +205,7 @@ def evaluate(
     for trace, dl_constructor in dataloaders.items():
         dataloader = dl_constructor()
         _evaluate_trace(dataloader, lightningmodule, device, trace, output)
-    logging.info(
-        f"Evaluation completed in {time.perf_counter() - start:.3f}s.")
+    logger.info(f"Evaluation completed in {time.perf_counter() - start:.3f}s.")
 
 if __name__ == "__main__":
     tyro.cli(evaluate)

src/nrdk/framework/lightningmodule.py

@@ -112,7 +112,7 @@ def compile(self) -> None:
         """
         jt_disable = os.environ.get("JAXTYPING_DISABLE", "0").lower()
         if jt_disable not in ("1", "true"):
-            self._log.error(
+            self._log.warning(
                 "torch.compile is currently incompatible with jaxtyping; "
                 "if you see type errors, set the environment variable "
                 "`JAXTYPING_DISABLE=1` to disable jaxtyping checks.")