From 836bb40fc7ca80c72634298b244d510b44f60627 Mon Sep 17 00:00:00 2001 From: Thomas Pinder Date: Sun, 8 Jan 2023 20:59:24 +0000 Subject: [PATCH] Remove unecessary imports --- jaxutils/parameters.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/jaxutils/parameters.py b/jaxutils/parameters.py index 3656024..c9fa665 100644 --- a/jaxutils/parameters.py +++ b/jaxutils/parameters.py @@ -1,14 +1,8 @@ -import warnings -from copy import deepcopy -from typing import Dict, Tuple +from typing import Dict from warnings import warn -import distrax as dx -import jax -import jax.numpy as jnp import jax.random as jr from jax.random import KeyArray -from jaxtyping import Array, Float from jaxutils import PyTree