First try to link with bayesnet

setup.py

@@ -5,6 +5,7 @@
 """
 
 from setuptools import Extension, setup
+from torch.utils import cpp_extension
 
 setup(
     ext_modules=[
@@ -20,5 +21,18 @@ setup(
                 "-std=c++17",
             ],
         ),
+        Extension(
+            name="bayesclass.cppBayesNetwork",
+            sources=[
+                "bayesclass/BayesNetwork.pyx",
+                "bayesclass/Network.cc",
+                "bayesclass/Node.cc",
+            ],
+            include_dirs=cpp_extension.include_paths(),
+            language="c++",
+            extra_compile_args=[
+                "-std=c++17",
+            ],
+        ),
     ]
 )