#858 add vectordot and state_vectordot to api docs

pybamm-team · Mar 14, 2020 · b341616 · b341616
1 parent 06ccfe8
commit b341616
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/docs/source/expression_tree/state_vector.rst b/docs/source/expression_tree/state_vector.rst
@@ -3,3 +3,6 @@ State Vector
 
 .. autoclass:: pybamm.StateVector
   :members:
+
+.. autoclass:: pybamm.StateVectorDot
+  :members:
diff --git a/docs/source/expression_tree/variable.rst b/docs/source/expression_tree/variable.rst
@@ -4,5 +4,9 @@ Variable
 .. autoclass:: pybamm.Variable
   :members:
 
+.. autoclass:: pybamm.VariableDot
+  :members:
+
 .. autoclass:: pybamm.ExternalVariable
   :members:
+
diff --git a/pybamm/expression_tree/symbol.py b/pybamm/expression_tree/symbol.py
@@ -486,10 +486,10 @@ def diff(self, variable):
         elif any(variable.id == x.id for x in self.pre_order()):
             return self._diff(variable)
         elif variable.id == pybamm.t.id and \
-            any(
-                isinstance(x, (pybamm.VariableBase, pybamm.StateVectorBase))
-                for x in self.pre_order()
-            ):
+                any(
+                    isinstance(x, (pybamm.VariableBase, pybamm.StateVectorBase))
+                    for x in self.pre_order()
+                ):
             return self._diff(variable)
         else:
             return pybamm.Scalar(0)