Quantcast
Channel: Recent Questions - Stack Overflow
Viewing all articles
Browse latest Browse all 12111

numba jitclass with record type of string

$
0
0

The v3 variable is string value. I could not run with below code which gives error.

import numpy as npimport pandas as pdfrom numba.experimental import jitclassfrom numba import typesimport osos.environ['NUMBA_VERBOSE'] = '1'# ----- BEGINNING OF THE MODIFIED PART ----- #recordType = types.Record([    ('v', {'type': types.int64, 'offset': 0, 'alignment': None, 'title': None}),    ('v2', {'type': types.float64, 'offset': 8, 'alignment': None, 'title': None}),    ('v3', {'type': types.bytes, 'offset': 16, 'alignment': None, 'title': None})], 32, False)spec = [    ('data', types.Array(recordType, 1, 'C', False))]# ----- END OF THE MODIFIED PART ----- #@jitclass(spec)class Test:    def __init__(self, data):        self.data = data    def loop(self):        v = self.data['v']        v2 = self.data['v2']        v3 = self.data['v3']        print("Inside loop:")        print("v:", v)        print("v2:", v2)        print("v3:", v3)# Create a dictionary with the datadata = {'v': [1, 2, 3], 'v2': [1.0, 2.0, 3.0], 'v3': ['a', 'b', 'c']}# Create the DataFramedf = pd.DataFrame(data)# Define the structured array dtypedtype = np.dtype([    ('v', np.int64),    ('v2', np.float64),    ('v3', 'S10')  # Byte string with maximum length of 10 characters])print(df.to_records(index=False))# Create the structured arraydata_array = np.array(list(df.to_records(index=False)), dtype=dtype)print("Original data array:")print(data_array)# Create an instance of the Test classtest = Test(data_array)test.loop()

Errors:

/home/totaljj/miniconda3/bin/conda run -n bt --no-capture-output python /home/totaljj/bt_lite_strategies/test/test_units/test_numba_obj.py Traceback (most recent call last):  File "/home/totaljj/bt_lite_strategies/test/test_units/test_numba_obj.py", line 13, in <module>    ('v3', {'type': types.bytes, 'offset': 16, 'alignment': None, 'title': None})AttributeError: module 'numba.core.types' has no attribute 'bytes'ERROR conda.cli.main_run:execute(124): `conda run python /home/totaljj/bt_lite_strategies/test/test_units/test_numba_obj.py` failed. (See above for error)Process finished with exit code 1,

Viewing all articles
Browse latest Browse all 12111

Trending Articles



<script src="https://jsc.adskeeper.com/r/s/rssing.com.1596347.js" async> </script>