Address feedback

nurse_scheduling.py

@@ -24,8 +24,16 @@
 from dimod import BinaryQuadraticModel
 from collections import defaultdict
 from copy import deepcopy
-import matplotlib.pyplot as plt
-from matplotlib.patches import Rectangle
+import matplotlib
+
+try:
+    import matplotlib.pyplot as plt
+    from matplotlib.patches import Rectangle
+except ImportError:
+    matplotlib.use("agg")
+    import matplotlib.pyplot as plt
+    from matplotlib.patches import Rectangle
+
 
 # Overall model variables: problem size
 # binary variable q_nd is the assignment of nurse n to day d
@@ -172,7 +180,6 @@ def get_nurse_and_day(index):
 
 # Get the results
 smpl = results.first.sample
-# energy = results.first.energy
 
 # Graphics
 print("\nBuilding schedule and checking constraints...\n")

requirements.txt

@@ -1 +1,4 @@
 dwave-ocean-sdk>=3.3.0
+
+matplotlib==3.3.4; python_version>'3.5'
+matplotlib==3.0.3; python_version=='3.5'

tests/test_nurse_scheduling.py

@@ -19,13 +19,7 @@
 example_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
 class IntegrationTests(unittest.TestCase):
-    @unittest.skipIf(os.getenv('SKIP_INT_TESTS'), "Skipping integration test.")
-    def test_smoke(self):
-        """run nurse_scheduling.py and check that nothing crashes"""
-
-        demo_file = os.path.join(example_dir, 'nurse_scheduling.py')
-        subprocess.check_output([sys.executable, demo_file])
-
+    @unittest.skipIf(os.getenv('SKIP_INT_TESTS'), "Skipping integration test.")    
     def test_schedule(self):
         file_path = os.path.join(example_dir, 'nurse_scheduling.py')