try fix tests

Author: SkafteNicki

requirements.txt

@@ -1,3 +1,3 @@
+torch>=1.7
 numpy>=1.16.4
-torchplot
 networkx

stochman/curves.py

@@ -81,8 +81,7 @@ def plot(
 
         """
         with torch.no_grad():
-            import torchplot as plt
-
+            import matplotlib.pyplot as plt
             t = torch.linspace(t0, t1, N, dtype=self.begin.dtype, device=self.device)
             points = self(t)  # NxD or BxNxD
 

tests/test_curves.py

@@ -122,7 +122,7 @@ def test_plotting_in_axis(self, curve_class):
         end = torch.randn(batch_size, dim)
         c = curve_class(begin, end, 20)
         try:
-            import torchplot as plt
+            import matplotlib.pyplot as plt
 
             fig, ax = plt.subplots(1, 1)
             c.plot(ax=ax)