print -> logging.

kaanaksit · Dec 3, 2023 · e6ab390 · e6ab390
1 parent e8a50a3
commit e6ab390
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/odak/visualize/plotly.py b/odak/visualize/plotly.py
@@ -1,11 +1,13 @@
 import sys
 import torch
+import logging
 try:
     import plotly
     import plotly.graph_objects as go
     from plotly.subplots import make_subplots
 except:
-    print('odak.visualize.plotly requires certain packages: pip install plotly kaleido')
+    warning = 'odak.visualize.plotly requires certain packages: pip install plotly kaleido'
+    logging.warning(warning)
 import numpy as np
 from ..wave import calculate_phase, calculate_amplitude, calculate_intensity