From 080099ecae3136511ce8d504feb400df31800928 Mon Sep 17 00:00:00 2001 From: Marc Sauter Date: Tue, 27 Feb 2024 12:26:34 +0100 Subject: [PATCH] isort --- .../training_recording/test_loss_ntk_recording_deployment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py b/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py index 33a4c60..d0032bb 100644 --- a/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py +++ b/CI/integration_tests/training_recording/test_loss_ntk_recording_deployment.py @@ -30,9 +30,9 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "-1" import numpy as np -from numpy.testing import assert_array_almost_equal import optax from neural_tangents import stax +from numpy.testing import assert_array_almost_equal from znnl.loss_functions import LPNormLoss from znnl.models import NTModel