fix: SqlRegistry _apply_object update statement (feast-dev#4042)

Nmroth42 · web-flow · commit ef62defbd801 · 2024-03-27T12:54:26.000-05:00
* fix sql registry update stmt

Signed-off-by: Nmroth42 &lt;mannoroth42@gmail.com&gt;

* refactor code

Signed-off-by: Nmroth42 &lt;mannoroth42@gmail.com&gt;

---------

Signed-off-by: Nmroth42 &lt;mannoroth42@gmail.com&gt;
diff --git a/sdk/python/feast/infra/registry/sql.py b/sdk/python/feast/infra/registry/sql.py
@@ -691,7 +691,10 @@ def _apply_object(
                 }
                 update_stmt = (
                     update(table)
-                    .where(getattr(table.c, id_field_name) == name)
+                    .where(
+                        getattr(table.c, id_field_name) == name,
+                        table.c.project_id == project,
+                    )
                     .values(
                         values,
                     )

Original file line number	Diff line number	Diff line change
`@@ -691,7 +691,10 @@ def _apply_object(`
`691`	`691`	`}`
`692`	`692`	`update_stmt = (`
`693`	`693`	`update(table)`
`694`		`- .where(getattr(table.c, id_field_name) == name)`
	`694`	`+ .where(`
	`695`	`+ getattr(table.c, id_field_name) == name,`
	`696`	`+ table.c.project_id == project,`
	`697`	`+ )`
`695`	`698`	`.values(`
`696`	`699`	`values,`
`697`	`700`	`)`