diff --git a/exhibit/core/generate/continuous.py b/exhibit/core/generate/continuous.py index 61b7b17..a2f8269 100644 --- a/exhibit/core/generate/continuous.py +++ b/exhibit/core/generate/continuous.py @@ -3,7 +3,6 @@ ''' # Standard library imports -import re # External library imports import numpy as np diff --git a/exhibit/core/generate/derived.py b/exhibit/core/generate/derived.py index 1f53b3b..5a49160 100644 --- a/exhibit/core/generate/derived.py +++ b/exhibit/core/generate/derived.py @@ -4,8 +4,6 @@ # Standard library imports import re -from datetime import time -from functools import partial # External library imports import numpy as np diff --git a/exhibit/core/generate/tests/test_continuous.py b/exhibit/core/generate/tests/test_continuous.py index 101d330..4da9f27 100644 --- a/exhibit/core/generate/tests/test_continuous.py +++ b/exhibit/core/generate/tests/test_continuous.py @@ -9,7 +9,6 @@ # External library imports import pandas as pd -from pandas.testing import assert_series_equal import numpy as np # Module under test diff --git a/exhibit/core/generate/tests/test_derived.py b/exhibit/core/generate/tests/test_derived.py index 0764ecb..7b4080b 100644 --- a/exhibit/core/generate/tests/test_derived.py +++ b/exhibit/core/generate/tests/test_derived.py @@ -4,7 +4,6 @@ # Standard library imports import unittest -from collections import namedtuple # External library imports import pandas as pd diff --git a/exhibit/core/generate/uuids.py b/exhibit/core/generate/uuids.py index 523b366..47779c4 100644 --- a/exhibit/core/generate/uuids.py +++ b/exhibit/core/generate/uuids.py @@ -54,6 +54,9 @@ def generate_uuid_column( for row in freq_df.itertuples(): + # initialise intermediate uuid list + _uuids = [] + # always round up the number of generated rows before casting to int _num_rows = int(np.ceil( num_rows * float(row.probability_vector) / int(row.frequency))) @@ -71,8 +74,6 @@ def generate_uuid_column( range_max = range_max + _num_rows continue - _uuids = [] - for _ in range(_num_rows): _uuids.append(uuid.UUID(int=rng.getrandbits(128), version=4).hex)