Merge pull request numenta#480 from htm-community/hotgym

Hotgym
breznak · Jun 7, 2019 · 5b0b4a9 · 5b0b4a9
2 parents a81757a + 5cf68cb
commit 5b0b4a9
Show file tree

Hide file tree

Showing 9 changed files with 4,659 additions and 6 deletions.
diff --git a/bindings/py/cpp_src/bindings/algorithms/py_SpatialPooler.cpp b/bindings/py/cpp_src/bindings/algorithms/py_SpatialPooler.cpp
@@ -308,6 +308,13 @@ using namespace sdr;
         py_SpatialPooler.def("getIterationLearnNum", &SpatialPooler::getIterationLearnNum);
 
 
+        py_SpatialPooler.def("__str__",
+            [](SpatialPooler &self) {
+                std::stringstream buf;
+                buf << self;
+                return buf.str(); });
+
+
         // pickle
 
         py_SpatialPooler.def(py::pickle(

diff --git a/bindings/py/cpp_src/bindings/algorithms/py_TemporalMemory.cpp b/bindings/py/cpp_src/bindings/algorithms/py_TemporalMemory.cpp
@@ -322,6 +322,12 @@ R"()");
         py_HTM.def_property_readonly("anomaly", [](const HTM_t &self) { return self.anomaly; },
           "Anomaly score updated with each TM::compute() call. "
         );
+
+        py_HTM.def("__str__",
+            [](HTM_t &self) {
+                std::stringstream buf;
+                buf << self;
+                return buf.str(); });
     }
 
 } // namespace nupic_ext
diff --git a/bindings/py/packaging/setup.py b/bindings/py/packaging/setup.py
@@ -254,6 +254,7 @@ def generateExtensions():
     install_requires=findRequirements(platform),
     package_data={
         "nupic.bindings": ["*.so", "*.pyd"],
+        "nupic.examples": ["*.csv"],
     },
     extras_require = {},
     zip_safe=False,