21 lines
920 B
Diff
21 lines
920 B
Diff
|
--- sympy/plotting/plot.py.orig 2016-03-08 11:35:33.000000000 -0700
|
||
|
+++ sympy/plotting/plot.py 2017-01-16 17:27:38.093969621 -0700
|
||
|
@@ -383,7 +383,7 @@ class Line2DBaseSeries(BaseSeries):
|
||
|
np = import_module('numpy')
|
||
|
c = self.line_color
|
||
|
if hasattr(c, '__call__'):
|
||
|
- f = np.vectorize(c)
|
||
|
+ f = np.vectorize(c, [np.float64])
|
||
|
arity = len(getargspec(c)[0])
|
||
|
if arity == 1 and self.is_parametric:
|
||
|
x = self.get_parameter_points()
|
||
|
@@ -700,7 +700,7 @@ class SurfaceBaseSeries(BaseSeries):
|
||
|
np = import_module('numpy')
|
||
|
c = self.surface_color
|
||
|
if isinstance(c, Callable):
|
||
|
- f = np.vectorize(c)
|
||
|
+ f = np.vectorize(c, [np.float64])
|
||
|
arity = len(getargspec(c)[0])
|
||
|
if self.is_parametric:
|
||
|
variables = list(map(centers_of_faces, self.get_parameter_meshes()))
|