From 561408575cb4531d20c57a23af1e67c534c86ee7 Mon Sep 17 00:00:00 2001
From: Johanna <johanna@zijderveld.de>
Date: Fri, 12 Apr 2024 16:37:40 +0200
Subject: [PATCH] fix imports to make pipeline happy

---
 profiling/graphene.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/profiling/graphene.py b/profiling/graphene.py
index 5b653bd..ddd3f63 100644
--- a/profiling/graphene.py
+++ b/profiling/graphene.py
@@ -3,6 +3,7 @@ import numpy as np
 from codes.model import Model
 from codes import kwant_examples
 from codes.kwant_helper import utils
+from codes.tb.utils import generate_guess
 from pyinstrument import Profiler
 import timeit
 import memray
@@ -16,7 +17,7 @@ nk = 600
 
 h_int = utils.builder_to_tb(int_builder, params)
 h_0 = utils.builder_to_tb(graphene_builder)
-guess = utils.generate_guess(frozenset(h_int), len(list(h_0.values())[0]))
+guess = generate_guess(frozenset(h_int), len(list(h_0.values())[0]))
 
 model = Model(h_0, h_int, filling)
 
-- 
GitLab