From 958f7114f1b31822dd5a39f2112015a396f69fe8 Mon Sep 17 00:00:00 2001
From: Itay Katav <ikatav@redhat.com>
Date: Wed, 5 Mar 2025 16:38:03 +0200
Subject: [PATCH] fix: added list value types

Signed-off-by: Itay Katav <ikatav@redhat.com>
---
 sdk/python/feast/type_map.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py
index 8e3941b05bf..bd9182d7618 100644
--- a/sdk/python/feast/type_map.py
+++ b/sdk/python/feast/type_map.py
@@ -580,6 +580,12 @@ def pa_to_feast_value_type(pa_type_as_str: str) -> ValueType:
             "bool": ValueType.BOOL,
             "null": ValueType.NULL,
             "list<element: double>": ValueType.DOUBLE_LIST,
+            "list<element: int64>": ValueType.INT64_LIST,
+            "list<element: int32>": ValueType.INT32_LIST,
+            "list<element: str>": ValueType.STRING_LIST,
+            "list<element: bool>": ValueType.BOOL_LIST,
+            "list<element: bytes>": ValueType.BYTES_LIST,
+            "list<element: float>": ValueType.FLOAT_LIST,
         }
         value_type = type_map[pa_type_as_str]