Virati/fix plotting

.devcontainer/Dockerfile

@@ -8,6 +8,11 @@ FROM mcr.microsoft.com/vscode/devcontainers/python:0-${VARIANT}
 ARG NODE_VERSION="none"
 RUN if [ "${NODE_VERSION}" != "none" ]; then su vscode -c "umask 0002 && . /usr/local/share/nvm/nvm.sh && nvm install ${NODE_VERSION} 2>&1"; fi
 
+
+USER vscode
+RUN curl -sSL https://install.python-poetry.org | python -
+ENV PATH = "${PATH}:/home/vscode/.local/bin"
+
 # [Optional] If your pip requirements rarely change, uncomment this section to add them to the image.
 # COPY requirements.txt /tmp/pip-tmp/
 # RUN pip3 --disable-pip-version-check --no-cache-dir install -r /tmp/pip-tmp/requirements.txt \

pyproject.toml

@@ -0,0 +1,18 @@
+[tool.poetry]
+name = "autodyn"
+version = "0.1.5"
+description = "Differentiable Dynamical Systems"
+authors = ["Vineet Tiruvadi <virati@gmail.com>"]
+license = "MIT License"
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.10"
+jax = "^0.4.25"
+matplotlib = "^3.8.3"
+networkx = "^3.2.1"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="autodyn",
-    version="0.1.4",
+    version="0.1.5",
     description="A module for differentiable dynamical systems",
     author="Vineet Tiruvadi",
     author_email="virati@gmail.com",

src/autodyn/core/dynamical.py

@@ -86,7 +86,7 @@ def plot_raster(self):
     def plot_phase(self):
         if self.D == 3:
             fig = plt.figure()
-            ax = fig.gca(projection="3d")
+            ax = fig.add_subplot(projection = '3d')
             ax.plot(self.raster[:, 0], self.raster[:, 1], self.raster[:, 2])
             plt.draw()
             plt.title("Phase Portrait")

src/autodyn/utils/functions.py

@@ -9,6 +9,5 @@ def unity(x):
 def quadratic(x: np.ndarray, params: List):
     return (x - params[0]) * (x - params[1])
 
-
 def rectify(x):
-    return np.maximum(x, 0)
+    return np.maximum(x, 0)