Skip to content

Commit

Permalink
pre-commits
Browse files Browse the repository at this point in the history
  • Loading branch information
drahc1R committed Jul 25, 2023
1 parent 3feb802 commit 47b0f26
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions synthetic_data/dataset_generator.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
"""Contains generator that returns collective df of requested distinct generators."""

import copy
from typing import List, Optional

import copy
import numpy as np
import pandas as pd
from numpy.random import Generator
Expand Down Expand Up @@ -106,7 +106,9 @@ def generate_dataset_by_class(
raise ValueError(f"generator: {data_type_var} is not a valid generator.")

col_generator_function = gen_funcs.get(data_type_var)
generated_data = col_generator_function(**col_, num_rows=dataset_length, rng=rng)
generated_data = col_generator_function(
**col_, num_rows=dataset_length, rng=rng
)
sort = col_.get("ordered", None)

if sort in ["ascending", "descending"]:
Expand Down

0 comments on commit 47b0f26

Please sign in to comment.