diff --git a/nemo/generators.py b/nemo/generators.py index b2e3148e..917ab471 100644 --- a/nemo/generators.py +++ b/nemo/generators.py @@ -259,6 +259,8 @@ def __init__(self, polygon, capacity, filename, column, label=None, assert np.all(~np.isnan(cls.csvdata)), \ f'Trace file {filename} contains NaNs; inspect file' cls.csvfilename = filename + # pylint limitation: https://github.com/pylint-dev/pylint/issues/9250 + # pylint: disable=unsubscriptable-object self.generation = cls.csvdata[::, column]