Skip to content

Commit

Permalink
Fix import warning
Browse files Browse the repository at this point in the history
  • Loading branch information
twomagpi committed Jul 2, 2020
1 parent b4b305d commit 5c10d9c
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
7 changes: 4 additions & 3 deletions pygom/model/_ode_composition.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
from functools import reduce

import sympy
from sympy.matrices import MatrixBase
import numpy as np

from .base_ode_model import BaseOdeModel
Expand Down Expand Up @@ -115,7 +116,7 @@ def generateDirectedDependencyGraph(ode_matrix, transition=None):
-1 and 1 to indicate the direction of the transition and the state.
All column sum to one, i.e. transition must have a source and target.
"""
assert isinstance(ode_matrix, sympy.matrices.MatrixBase), \
assert isinstance(ode_matrix, MatrixBase), \
"Expecting a vector of expressions"

if transition is None:
Expand Down Expand Up @@ -150,7 +151,7 @@ def getUnmatchedExpressionVector(expr_vec, full_output=False):
list:
of unmatched expressions, i.e. birth or death processes
"""
assert isinstance(expr_vec, sympy.matrices.MatrixBase), \
assert isinstance(expr_vec, MatrixBase), \
"Expecting a vector of expressions"

transition = reduce(lambda x, y: x + y, map(getExpressions, expr_vec))
Expand Down Expand Up @@ -181,7 +182,7 @@ def getMatchingExpressionVector(expr_vec, outTuple=False):
list:
of matched expressions, i.e. transitions
"""
assert isinstance(expr_vec, sympy.matrices.MatrixBase), \
assert isinstance(expr_vec, MatrixBase), \
"Expecting a vector of expressions"

transition = list()
Expand Down
1 change: 1 addition & 0 deletions pygom/model/simulate.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

import numpy as np
import sympy
import sympy.matrices.matrices
import scipy.stats

from .deterministic import DeterministicOde
Expand Down

0 comments on commit 5c10d9c

Please sign in to comment.