[plotting] remove matplotlib as global dependency

Author: nikohansen

comocma/como_logger.py

@@ -1,7 +1,6 @@
 """Defines the `COMOPlot` class which can store (via callback) and plot data.
 """
 import numpy as np
-import matplotlib.pyplot as plt
 import os
 import datetime
 from time import time
@@ -343,6 +342,7 @@ def plot(self):
 
     def plot_proportion_dominated_final_incumbents(self):
         """Plot the proportion of dominated final incumbents."""
+        import matplotlib.pyplot as plt
         dic = self.load()
         try:
             n_runs = len(dic["completed_runs"])
@@ -362,6 +362,7 @@ def plot_iterations_per_restart(self):
         """
         Plot the number of iterations per restart and the condition number.
         """
+        import matplotlib.pyplot as plt
         dic = self.load()
         try:
             n_runs = len(dic["completed_runs"])
@@ -390,6 +391,7 @@ def plot_iterations_per_restart(self):
 
     def plot_convergence_speed(self):
         """Plot the convergence speed."""
+        import matplotlib.pyplot as plt
         dic = self.load()
         if self.offset is None:
             offset = dic["hv_archive"][-1]
@@ -409,6 +411,7 @@ def plot_convergence_speed(self):
 
     def plot_archive(self):
         """Plot the archive."""
+        import matplotlib.pyplot as plt
         dic = self.load()
         non_dominated_kernels = [v for v in dic["objective_values"][-1]
                                  if v in dic["last_archive"]]
@@ -436,6 +439,7 @@ def plot_archive(self):
 
     def plot_hvi(self):
         """Plot information regarding hypervolume improvement."""
+        import matplotlib.pyplot as plt
         dic = self.load()
         try:
             n_runs = len(dic["completed_runs"])
@@ -480,6 +484,7 @@ def plot_stepsizes(self):
         Plot the first, the last, the minimum and the maximum stepsizes of all completed runs, plus 
         the initial stepsize of the last run.
         """
+        import matplotlib.pyplot as plt
         dic = self.load()
 
         try:
@@ -511,6 +516,7 @@ def plot_uhvi(self):
         * correct the bug where there is only one tick if the span is too little, making the plot
         unreadable 
         """
+        import matplotlib.pyplot as plt
         dic = self.load()
 
         try:

comocma/sofomore_logger.py

@@ -5,7 +5,7 @@
 import cma
 from cma import interfaces
 import os
-import matplotlib.pyplot as plt
+# import matplotlib.pyplot as plt  # don't require matplotlib in tests
 import ast
 import warnings
 
@@ -463,6 +463,7 @@ def plot(self, filename=None, x_iteration = False):
     def plot_front(self, aspect=None):
         """
         """
+        from matplotlib import pyplot as plt
         if aspect is not None:
             myaxes = plt.gca()
             try: