Skip to content

Commit d8e4321

Browse files
authored
Embedding lists (#58)
* Submodule change * Submodule change * Merge wiht dev * Submodule change
1 parent 7b951b6 commit d8e4321

File tree

3 files changed

+10
-3
lines changed

3 files changed

+10
-3
lines changed

controller.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,14 @@ def run_encoding(
178178
additional_data: Optional[Any] = None,
179179
) -> int:
180180
session_token = general.get_ctx_token()
181-
initial_count = record.count(project_id)
181+
initial_count = 0
182+
if (
183+
attribute.get_by_name(project_id, attribute_name).data_type
184+
== enums.DataTypes.EMBEDDING_LIST.value
185+
):
186+
initial_count = record.count_attribute_list_entries(project_id, attribute_name)
187+
else:
188+
initial_count = record.count(project_id)
182189
seed_str = embedding_name
183190
torch.manual_seed(zlib.adler32(bytes(seed_str, "utf-8")))
184191
notification.create(

submodules/s3

Submodule s3 updated 1 file

0 commit comments

Comments
 (0)