Mercurial > hg > tvii

changeset 19:cd43ce453358

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
[numpy] of course we have to cast everything here too
author Jeff Hammel <k0scist@gmail.com>
date Sun, 03 Sep 2017 20:46:26 -0700
parents e92fc4a17336
children d6d2ecb33c95
files tests/test_sigmoid.py tvii/sigmoid.py
diffstat 2 files changed, 4 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- a/tests/test_sigmoid.py	Sun Sep 03 19:40:29 2017 -0700
+++ b/tests/test_sigmoid.py	Sun Sep 03 20:46:26 2017 -0700
@@ -15,8 +15,7 @@
         """test two points of the sigmoid function"""
 
         answer = sigmoid([0,2])
-        assert len(answer(2))
-        
+        assert len(answer) == 2
 
 if __name__ == '__main__':
     unittest.main()
--- a/tvii/sigmoid.py	Sun Sep 03 19:40:29 2017 -0700
+++ b/tvii/sigmoid.py	Sun Sep 03 20:46:26 2017 -0700
@@ -3,8 +3,10 @@
 """
 
 import numpy as np
-
+from .iterable import isiterable
 
 def sigmoid(z):
     """https://en.wikipedia.org/wiki/Sigmoid_function"""
+    if not isinstance(z, np.ndarray) and isiterable(z):
+        z = np.array(z)
     return 1./(1. + np.exp(-z))