Fix MlModel statement usage

Oyelowo · Jan 18, 2024 · 397671a · 397671a
1 parent 239d7c7
commit 397671a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/migrator/src/database/queries.rs b/migrator/src/database/queries.rs
@@ -274,7 +274,7 @@ impl DefineStatementRaw {
                     };
                     remove_stmt.to_raw().build()
                 }
-                DefineStatement::MlModel(ml) => remove_model(get_resource_name(ml.name.to_raw()))
+                DefineStatement::Model(ml) => remove_model(get_resource_name(ml.name.to_raw()))
                     .version(ml.version)
                     .to_raw()
                     .build(),