account for modshogun -> shogun name change

Author: djsutherland

two_sample/fixed_run.py

@@ -6,7 +6,11 @@
 import os
 import sys
 
-import modshogun as sg
+try:
+    import modshogun as sg
+except ImportError:  # new versions just call it shogun
+    import shogun as sg
+
 import numpy as np
 import pandas as pd
 import progressbar as pb

two_sample/mmd_test.py

@@ -9,7 +9,10 @@
 
 import numpy as np
 from scipy import linalg, stats
-import modshogun as sg
+try:
+    import modshogun as sg
+except ImportError:  # new versions just call it shogun
+    import shogun as sg
 
 if 'OMP_NUM_THREADS' in os.environ:
     num_threads = int(os.environ['OMP_NUM_THREADS'])