fix imports

Author: ArthurDanjou

histogram.py

@@ -19,6 +19,8 @@
 
 import matplotlib.pyplot as plt
 
+import pandas as pd
+
 from krum import aggregators, tools
 
 # Change common font for the default LaTeX one
@@ -62,7 +64,7 @@ def gtk_main():
                 gtk_lazy_main = thread
         # Submit the job to the main loop
         GLib.idle_add(closure)
-except Exception:
+except Exception as err:
 
     def gtk_run(closure):
         """Sink in case GTK cannot be used.
@@ -220,7 +222,7 @@ def __init__(self, path_results):
         # Load training data
         path_study = path_results / "study"
         try:
-            data_study = pandas.read_csv(
+            data_study = pd.read_csv(
                 path_study, sep="\t", index_col=0, na_values="     nan"
             )
             data_study.index.name = "Step number"
@@ -232,7 +234,7 @@ def __init__(self, path_results):
         # Load evaluation data
         path_eval = path_results / "eval"
         try:
-            data_eval = pandas.read_csv(path_eval, sep="\t", index_col=0)
+            data_eval = pd.read_csv(path_eval, sep="\t", index_col=0)
             data_eval.index.name = "Step number"
         except Exception as err:
             tools.warning(
@@ -457,7 +459,7 @@ def include(self, data, *cols, errs=None, lalp=1.0, ccnt=None):
         # Recover the dataframe if a session was given
         if isinstance(data, Session):
             data = data.data
-        elif not isinstance(data, pandas.DataFrame):
+        elif not isinstance(data, pd.DataFrame):
             raise RuntimeError(
                 f"Expected a Session or DataFrame for 'data', got a {tools.fullqual(type(data))!r}"
             )
@@ -531,7 +533,7 @@ def include_single(self, data, key, col, err=None, lalp=1.0, ccnt=None):
         # Recover the dataframe if a session was given
         if isinstance(data, Session):
             data = data.data
-        elif not isinstance(data, pandas.DataFrame):
+        elif not isinstance(data, pd.DataFrame):
             raise RuntimeError(
                 f"Expected a Session or DataFrame for 'data', got a {tools.fullqual(type(data))!r}"
             )
@@ -719,7 +721,7 @@ def include(self, data):
           self
         """
         # Convert 'pandas.Series' to numpy
-        if isinstance(data, pandas.Series):
+        if isinstance(data, pd.Series):
             data = data.to_numpy()
         # Make the histogram
         self._ax.hist(data, bins=self._bins)

krum/experiments/__init__.py

@@ -46,7 +46,7 @@
 
 import pathlib
 
-import tools
+from krum import tools
 
 # ---------------------------------------------------------------------------- #
 # Load all local modules