Fix entity column exclusion in codegen and type widening in mapper

YassinNouh21 · YassinNouh21 · commit 4318134066cd · 2026-02-07T14:28:50.000+01:00
- codegen: stop excluding entity columns from FeatureView schema,
  matching mapper behavior (FeatureView.__init__ expects them)
- mapper: map Int32 to ValueType.INT32 and Float32 to ValueType.FLOAT
  instead of widening to INT64/DOUBLE
diff --git a/sdk/python/feast/dbt/codegen.py b/sdk/python/feast/dbt/codegen.py
@@ -248,7 +248,9 @@ def generate(
         if not entity_cols:
             raise ValueError("At least one entity column must be specified")
 
-        excluded = set(entity_cols) | {self.timestamp_field}
+        # Note: entity columns should NOT be excluded - FeatureView.__init__
+        # expects entity columns to be in the schema and will extract them
+        excluded = {self.timestamp_field}
         if exclude_columns:
             excluded.update(exclude_columns)
 
diff --git a/sdk/python/feast/dbt/mapper.py b/sdk/python/feast/dbt/mapper.py
@@ -29,9 +29,9 @@
 # Mapping from FeastType to ValueType for entity value inference
 FEAST_TYPE_TO_VALUE_TYPE: Dict[FeastType, ValueType] = {
     String: ValueType.STRING,
-    Int32: ValueType.INT64,
+    Int32: ValueType.INT32,
     Int64: ValueType.INT64,
-    Float32: ValueType.DOUBLE,
+    Float32: ValueType.FLOAT,
     Float64: ValueType.DOUBLE,
     Bool: ValueType.BOOL,
     Bytes: ValueType.BYTES,
