From 7a44d4fa54747040dc5d29ee8899bb5998230cc6 Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Thu, 11 May 2023 08:29:55 +0200
Subject: [PATCH] Small Fix

---
 pystencils_benchmark/benchmark.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pystencils_benchmark/benchmark.py b/pystencils_benchmark/benchmark.py
index 67cc3d5..34bdf6f 100644
--- a/pystencils_benchmark/benchmark.py
+++ b/pystencils_benchmark/benchmark.py
@@ -6,9 +6,9 @@ from jinja2 import Environment, PackageLoader, StrictUndefined
 import numpy as np
 
 from pystencils.backends.cbackend import generate_c, get_headers
-from pystencils.astnodes import KernelFunction, PragmaBlock
+from pystencils.astnodes import KernelFunction
 from pystencils.enums import Backend
-from pystencils.data_types import get_base_type
+from pystencils.typing import get_base_type
 from pystencils.sympyextensions import prod
 from pystencils.integer_functions import modulo_ceil
 
-- 
GitLab