fix imports to make pipeline happy

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)