andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Fix traces table filtering for string arguments

Browse source
This commit is contained in:
Andras Schmelczer 2022-08-11 16:49:19 +02:00
parent 83e111f088
commit 49968691cc
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
2 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
great_ai/persistence/parallel_tinydb_driver.py
View file

@ -67,7 +67,7 @@ class ParallelTinyDbDriver(TracingDatabaseDriver):
if not documents:
return [], 0
df = pd.DataFrame(documents)
df = pd.DataFrame([Trace.parse_obj(d).to_flat_dict() for d in documents])
for f in conjunctive_filters:
operator = f.operator.lower()
@ -76,7 +76,12 @@ class ParallelTinyDbDriver(TracingDatabaseDriver):
getattr(df[f.property], operator_mapping[f.operator])(f.value)
]
elif operator == "contains":
df = df.loc[df[f.property].str.contains(f.value, case=False)]
df = df.loc[
df[f.property].str.contains(
str(int(f.value)) if isinstance(f.value, float) else f.value,
case=False,
)
]
if sort_by:
df.sort_values(