From a6f987d346937730762a1420615cd40c6c405f6e Mon Sep 17 00:00:00 2001
From: Brady Planden <brady.planden@gmail.com>
Date: Wed, 4 Dec 2024 17:37:31 +0000
Subject: [PATCH] refactor: update init structure

---
 pybop/__init__.py              | 2 +-
 pybop/experimental/__init__.py | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)
 delete mode 100644 pybop/experimental/__init__.py

diff --git a/pybop/__init__.py b/pybop/__init__.py
index 11ae8901e..c4c9d1611 100644
--- a/pybop/__init__.py
+++ b/pybop/__init__.py
@@ -120,7 +120,7 @@
 #
 # Experimental
 #
-from .experimental import BaseJaxCost, JaxSumSquaredError, JaxLogNormalLikelihood, JaxGaussianLogLikelihoodKnownSigma
+from .experimental.jax_costs import BaseJaxCost, JaxSumSquaredError, JaxLogNormalLikelihood, JaxGaussianLogLikelihoodKnownSigma
 
 #
 # Optimiser classes
diff --git a/pybop/experimental/__init__.py b/pybop/experimental/__init__.py
deleted file mode 100644
index 67b91072a..000000000
--- a/pybop/experimental/__init__.py
+++ /dev/null
@@ -1 +0,0 @@
-from .jax_costs import BaseJaxCost, JaxLogNormalLikelihood, JaxSumSquaredError, JaxGaussianLogLikelihoodKnownSigma