From 6afe6463c2f737fad2aab39c01e4f5a3732e29b3 Mon Sep 17 00:00:00 2001
From: Alejandro Saucedo <axsauze@gmail.com>
Date: Sun, 1 Nov 2020 16:27:59 +0000
Subject: [PATCH] Updated to add opbase

---
 python/src/main.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/python/src/main.cpp b/python/src/main.cpp
index 5fe74f021..e3b7fb371 100644
--- a/python/src/main.cpp
+++ b/python/src/main.cpp
@@ -21,7 +21,10 @@ PYBIND11_MODULE(komputepy, m) {
         .def(py::init(
             [](const std::vector<float>& data, kp::Tensor::TensorTypes tensorTypes) {
                 return std::unique_ptr<kp::Tensor>(new kp::Tensor(data, tensorTypes));
-            }));
+            }))
+        .def("data", &kp::Tensor::data);
+
+    py::class_<kp::OpBase>(m, "OpBase");
 
 #ifdef VERSION_INFO
     m.attr("__version__") = VERSION_INFO;