Skip to content

Commit bd1c1e9

Browse files
committed
remove all PKArray usage
1 parent 9ec995c commit bd1c1e9

2 files changed

Lines changed: 2 additions & 28 deletions

File tree

pykokkos/interface/views.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from typing import Dict, Generic, Iterator, List, Optional, Tuple, TypeVar, Union
1010

1111
import numpy as np
12-
from pykokkos.lib.pk_array import PKArray
1312

1413
import pykokkos as pk
1514
from pykokkos.bindings import kokkos
@@ -539,7 +538,7 @@ def __eq__(self, other):
539538
else:
540539
raise ValueError("unexpected types!")
541540
result_np = np.equal(np.array(self), np.array(new_other))
542-
return PKArray(result_np)
541+
return result_np
543542

544543
def __hash__(self):
545544
try:
@@ -704,7 +703,7 @@ def __eq__(self, other):
704703
else:
705704
raise ValueError("unexpected types!")
706705
result_np = np.equal(np.array(self), np.array(new_other))
707-
return PKArray(result_np)
706+
return result_np
708707

709708
def __add__(self, other):
710709
if isinstance(other, float):

pykokkos/lib/pk_array.py

Lines changed: 0 additions & 25 deletions
This file was deleted.

0 commit comments

Comments
 (0)