Skip to content

Commit

Permalink
feat: add column characteristic enum (#43)
Browse files Browse the repository at this point in the history
  • Loading branch information
alexbarros authored Sep 25, 2024
1 parent 8654e01 commit 773fe46
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/datascience/ydata/datascience/common/__init__.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,10 @@
from .privacy import PrivacyLevel
from .anonymizer import AnonymizerType
from .column_characteristic import ColumnCharacteristic


__all__ = [
"PrivacyLevel",
"AnonymizerType"
"AnonymizerType",
"ColumnCharacteristic"
]
22 changes: 22 additions & 0 deletions src/datascience/ydata/datascience/common/column_characteristic.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
from ydata.core.enum import StringEnum


class ColumnCharacteristic(StringEnum):
# Adding new characteristics may require adding a new generator to:
# src/ydatasynthesizers/ydata-synthesizers/ydata/synthesizers/faker/utils.py
ID = "id"
EMAIL = "email"
URL = "url"
UUID = "uuid"
NAME = "name"
PHONE = "phone"
VAT = "vat"
IBAN = "iban"
CREDIT_CARD = "credit_card"
COUNTRY = "country"
ZIPCODE = "zipcode"
ADDRESS = "address"
PII = 'PII'
# Generic characteristic grouping Country, Address, etc.
LOCATION = 'location'
PERSON = 'person' # Generic characteristic grouping person related attributes

0 comments on commit 773fe46

Please sign in to comment.