From 35db19ffd0626341ddf323b201a33275c6e654bc Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 21 Jan 2020 08:55:39 +0100
Subject: [PATCH] Add typeannotations and docstring for KernelWrapper
---
pystencils/kernel_wrapper.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/pystencils/kernel_wrapper.py b/pystencils/kernel_wrapper.py
index 0e327711e..3494b52a9 100644
--- a/pystencils/kernel_wrapper.py
+++ b/pystencils/kernel_wrapper.py
@@ -1,11 +1,14 @@
-"""
-Light-weight wrapper around a compiled kernel
-"""
import pystencils
class KernelWrapper:
- def __init__(self, kernel, parameters, ast_node):
+ """
+ Light-weight wrapper around a compiled kernel.
+
+ Can be called while still providing access to underlying AST.
+ """
+
+ def __init__(self, kernel, parameters, ast_node: pystencils.astnodes.KernelFunction):
self.kernel = kernel
self.parameters = parameters
self.ast = ast_node
--
GitLab