Merge pull request #2 from fAndreuzzi/patch-1

Use `np.polynomial.chebyshev.chebroots` instead of `np.roots`

pina/chebyshev.py

@@ -1,6 +1,7 @@
-from mpmath import chebyt, chop, taylor
 import numpy as np
 def chebyshev_roots(n):
     """ Return the roots of *n* Chebyshev polynomials (between [-1, 1]) """
-    return np.roots(chop(taylor(lambda x: chebyt(n, x), 0, n))[::-1])
+    coefficents = np.zeros(n+1)
+    coefficents[-1] = 1
+    return np.polynomial.chebyshev.chebroots(coefficents)
 

setup.py

@@ -14,8 +14,7 @@ VERSION = meta['__version__']
 KEYWORDS = 'physics-informed neural-network'
 
 REQUIRED = [
-    'future', 'numpy', 'matplotlib', 'torch', 'mpmath'
-    
+    'future', 'numpy', 'matplotlib', 'torch'  
 ]
 
 EXTRAS = {