From 46c7187785e9b2427f4e46d3634773a0f8a8c8a4 Mon Sep 17 00:00:00 2001 From: michaeldeistler Date: Wed, 6 Dec 2023 15:42:00 +0100 Subject: [PATCH] isort --- jaxley/modules/network.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/jaxley/modules/network.py b/jaxley/modules/network.py index e528230e..0a917e3e 100644 --- a/jaxley/modules/network.py +++ b/jaxley/modules/network.py @@ -1,14 +1,12 @@ import itertools from copy import deepcopy -from typing import Callable, Dict, List, Tuple, Optional, Union - +from typing import Callable, Dict, List, Optional, Tuple, Union import jax.numpy as jnp +import networkx as nx import numpy as np import pandas as pd from jax import vmap -import networkx as nx - from jaxley.connection import Connectivity from jaxley.modules.base import Module, View