Browse Source

Wrap pandas import to avoid warnings

Signed-off-by: Vadim Markovtsev <vadim@sourced.tech>
Vadim Markovtsev 6 years ago
parent
commit
79a0cc27c0
1 changed files with 13 additions and 3 deletions
  1. 13 3
      labours.py

+ 13 - 3
labours.py

@@ -551,8 +551,18 @@ def interpolate_burndown_matrix(matrix, granularity, sampling):
     return daily
     return daily
 
 
 
 
-def load_burndown(header, name, matrix, resample):
+def import_pandas():
     import pandas
     import pandas
+    try:
+        from pandas.plotting import register_matplotlib_converters
+        register_matplotlib_converters()
+    except ImportError:
+        pass
+    return pandas
+
+
+def load_burndown(header, name, matrix, resample):
+    pandas = import_pandas()
 
 
     start, last, sampling, granularity = header
     start, last, sampling, granularity = header
     assert sampling > 0
     assert sampling > 0
@@ -621,7 +631,7 @@ def load_burndown(header, name, matrix, resample):
 
 
 
 
 def load_ownership(header, sequence, contents, max_people):
 def load_ownership(header, sequence, contents, max_people):
-    import pandas
+    pandas = import_pandas()
 
 
     start, last, sampling, _ = header
     start, last, sampling, _ = header
     start = datetime.fromtimestamp(start)
     start = datetime.fromtimestamp(start)
@@ -1464,7 +1474,7 @@ def main():
 
 
     def run_times():
     def run_times():
         rt = reader.get_run_times()
         rt = reader.get_run_times()
-        import pandas
+        pandas = import_pandas()
         series = pandas.to_timedelta(pandas.Series(rt).sort_values(ascending=False), unit="s")
         series = pandas.to_timedelta(pandas.Series(rt).sort_values(ascending=False), unit="s")
         df = pandas.concat([series, series / series.sum()], axis=1)
         df = pandas.concat([series, series / series.sum()], axis=1)
         df.columns = ["time", "ratio"]
         df.columns = ["time", "ratio"]