From f61d53f5e075781bf6e265dd268124bf58f2110c Mon Sep 17 00:00:00 2001 From: michaeldeistler Date: Wed, 6 Dec 2023 18:17:58 +0100 Subject: [PATCH] isort --- jaxley/modules/network.py | 3 ++- jaxley/utils/plot_utils.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/jaxley/modules/network.py b/jaxley/modules/network.py index 7d5190b8..6ad6b093 100644 --- a/jaxley/modules/network.py +++ b/jaxley/modules/network.py @@ -1,8 +1,9 @@ import itertools from copy import deepcopy from typing import Callable, Dict, List, Optional, Tuple, Union -import matplotlib.pyplot as plt + import jax.numpy as jnp +import matplotlib.pyplot as plt import networkx as nx import numpy as np import pandas as pd diff --git a/jaxley/utils/plot_utils.py b/jaxley/utils/plot_utils.py index 80ca052c..d972de5e 100644 --- a/jaxley/utils/plot_utils.py +++ b/jaxley/utils/plot_utils.py @@ -68,7 +68,8 @@ def plot_morph( counter_highlight_branches = 0 lines = [] - fig, ax = plt.subplots(1, 1, figsize=figsize) + if fig is None or ax is None: + fig, ax = plt.subplots(1, 1, figsize=figsize) for b in range(cell.total_nbranches): if parents[b] > -1: