multimodalart HF Staff commited on
Commit
defb15d
·
verified ·
1 Parent(s): f7e2c07

Make IPython import optional

Browse files
Files changed (1) hide show
  1. dataloaders/pymo/viz_tools.py +4 -1
dataloaders/pymo/viz_tools.py CHANGED
@@ -1,7 +1,10 @@
1
  import pandas as pd
2
  import numpy as np
3
  import matplotlib.pyplot as plt
4
- import IPython
 
 
 
5
  import os
6
 
7
  def save_fig(fig_id, tight_layout=True):
 
1
  import pandas as pd
2
  import numpy as np
3
  import matplotlib.pyplot as plt
4
+ try:
5
+ import IPython
6
+ except ImportError:
7
+ IPython = None # Only needed for notebook display
8
  import os
9
 
10
  def save_fig(fig_id, tight_layout=True):