Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
matthiasdiener committed Oct 9, 2024
1 parent 70a6887 commit 4f9f95f
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions pytato/analysis/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -581,12 +581,14 @@ def update_for_ndarray(self, key_hash: Any, key: Any) -> None:
self.rec(key_hash, key.data.tobytes())

def update_for_TaggableCLArray(self, key_hash: Any, key: Any) -> None:
from arraycontext.impl.pyopencl.taggable_cl_array import TaggableCLArray
from arraycontext.impl.pyopencl.taggable_cl_array import (
TaggableCLArray, # pylint: disable=import-error
)
assert isinstance(key, TaggableCLArray)
self.rec(key_hash, key.get())

def update_for_Array(self, key_hash: Any, key: Any) -> None:
from pyopencl import Array
from pyopencl.array import Array
assert isinstance(key, Array)
self.rec(key_hash, key.get())

Expand Down

0 comments on commit 4f9f95f

Please sign in to comment.