Merge branch 'JBjoernskov:main' into main

README.md

@@ -49,6 +49,11 @@ To be added soon.
 
 To be added soon.
 
+### Neural Policy Controller
+
++ <a target="_blank" href="https://colab.research.google.com/github/JBjoernskov/Twin4Build/blob/main/twin4build/examples/neural_policy_controller_example/neural_policy_example.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a> Part 1: Training a neural policy controller for the space model
+
 ## Documentation
 The documentation can be found [online](https://twin4build.readthedocs.io/en/latest/index.html).
 Below is a code snippet showing the basic functionality of the package.

setup.py

@@ -12,7 +12,7 @@
     platforms=["Windows", "Linux"],
     packages=setuptools.find_packages(),
     include_package_data=True,
-    package_data={'': ['*.pickle', '*.xlsx', "*.pt", "*.fmu", "*.ini", "*.xlsm", "*.csv"]},
+    package_data={'': ['*.pickle', '*.xlsx', "*.pt", "*.fmu", "*.ini", "*.xlsm", "*.csv", "*.json", "*.pth"]},
     install_requires=[
         "matplotlib",
         "networkx",