aboutsummaryrefslogtreecommitdiff
path: root/test_utils/models/upk
diff options
context:
space:
mode:
authorGreg Stupp2026-09-18 22:39:18 +0000
committerGreg Stupp2026-09-18 22:39:18 +0000
commitb373b992cae3a9628a9cf64387c7b8e286d6e1e0 (patch)
tree5e21759ea2c6925e8f54328fbbac88b6be007648 /test_utils/models/upk
parent32f30b5ddc600e22e071f7659f1872a18ebf42c7 (diff)
parente3c6a824f4176402e40217c5ce70f837dedc5c3d (diff)
downloadgeneralresearch-b373b992cae3a9628a9cf64387c7b8e286d6e1e0.tar.gz
generalresearch-b373b992cae3a9628a9cf64387c7b8e286d6e1e0.zip
Merges pull request #4 v3.5.0
User Cashout Stuff pulled out of grpc, user ip history enrichments with grip (no maxmind)
Diffstat (limited to 'test_utils/models/upk')
-rw-r--r--test_utils/models/upk/conftest.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/test_utils/models/upk/conftest.py b/test_utils/models/upk/conftest.py
index ad96bbb..520d566 100644
--- a/test_utils/models/upk/conftest.py
+++ b/test_utils/models/upk/conftest.py
@@ -33,6 +33,13 @@ def insert_data_from_csv(
df["id"] = df["id"].map(lambda x: UUID(x).hex)
args = df.to_dict("tight")["data"]
+ # Fix issue with nullable cols
+ if table_name == 'marketplace_question':
+ for arg in args:
+ for idx, col in enumerate(arg):
+ if pd.isnull(col):
+ arg[idx] = None
+
with thl_web_rw.make_connection() as conn:
with conn.cursor() as c:
if disable_fk_checks: