From 0dbf103e064754f6bef3191a1ca38a8ab8d3f937 Mon Sep 17 00:00:00 2001
From: Henry Korb <henry.korb@geo.uu.se>
Date: Tue, 21 Mar 2023 18:18:56 +0100
Subject: [PATCH] minor clean up in binding stubs

---
 pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi | 1 +
 pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi    | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi b/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi
index a41b7934c..a8b7ae9e7 100644
--- a/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi
+++ b/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi
@@ -36,6 +36,7 @@ from typing import ClassVar
 
 from typing import overload
 
+from pyfluids.bindings.basics import logger as logger
 class ConfigurationFile:
     def __init__(self) -> None: ...
     def contains(self, key: str) -> bool: ...
diff --git a/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi b/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
index 36c2fea76..41b019960 100644
--- a/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
+++ b/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
@@ -37,8 +37,9 @@ from typing import Any, Callable, ClassVar, List, Optional
 from typing import overload
 import numpy
 import pyfluids.bindings.basics
-import pyfluids.bindings.gpu.grid_generator as grid_generator
 
+from  pyfluids.bindings.gpu import grid_generator as grid_generator
+from pyfluids.bindings.gpu import probes as probes
 class ActuatorFarm(PreCollisionInteractor):
     def __init__(self, number_of_blades_per_turbine: int, density: float, number_of_nodes_per_blade: int, epsilon: float, level: int, delta_t: float, delta_x: float, use_host_arrays: bool) -> None: ...
     def add_turbine(self, posX: float, posY: float, posZ: float, diameter: float, omega: float, azimuth: float, yaw: float, bladeRadii: List[float]) -> None: ...
-- 
GitLab