Commit ec625b05 authored by hazrmard's avatar hazrmard
Browse files

standard.py: 0 indexing behavior when simulating.

parent c95e806c

pystatespace/neural.py

+1 −0
Changes for pystatespace/neural.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -106,6 +106,7 @@ class Trapezoid(nn.Module): 
        Returns:

            np.ndarray -- array of shape `N x Outputs`,

        """

        # pylint: disable=not-callable

        self._init(x0, u)

        xdim = self.dims

        if isinstance(t, Number):

pystatespace/standard.py

+2 −2
Changes for pystatespace/standard.py: 2 added lines, 2 removed lines.
Original line number Diff line number Diff line
@@ -67,7 +67,7 @@ class Trapezoid: 
        # t:    0   1   2   3   4   5

        # x:    x0  (                )

        # u:    u   u   u   u   u   u

        #

        # y:    y   y   y   y   y   y

        xdim = self.dims

        if isinstance(t, Number):

            t = np.linspace(0, t, len(u) + 1, endpoint=True)
@@ -140,7 +140,7 @@ class SS_ODE(Trapezoid): 
        -> Tuple[np.ndarray, np.ndarray]:

        if isinstance(t, Number):

            t = np.linspace(0, t, len(u), endpoint=False)

        t = np.concatenate(([t[0]], t), axis=0)

        t = np.concatenate(([t[0]], t), axis=0)  # len(t) increased by 1 now

        u = np.concatenate(([u[0]-u[0]], u), axis=0)

        X = np.zeros((len(t), self.dims))

        Y = np.zeros((len(t), self.outputs))

pystatespace/test.py

+1 −3
Changes for pystatespace/test.py: 1 added line, 3 removed lines.
Original line number Diff line number Diff line
@@ -2,13 +2,11 @@ 
Unit tests for code.

"""







from unittest import TestCase, main



import numpy as np



from .state_space import Trapezoid, LTISystem, SS_ODE

from .standard import Trapezoid, LTISystem, SS_ODE