qualify schema namespace

montanalow · montanalow · commit afa820fcc28c · 2023-03-15T10:38:54.000-07:00
diff --git a/pgml-extension/src/orm/dataset.rs b/pgml-extension/src/orm/dataset.rs
@@ -98,7 +98,7 @@ fn drop_table_if_exists(table_name: &str) {
         (PgBuiltInOids::TEXTOID.oid(), table_name.clone().into_datum())
     ]).unwrap().unwrap();
     match table_count {
-        1 => Spi::run(&format!(r#"DROP TABLE {table_name}"#)).unwrap(),
+        1 => Spi::run(&format!(r#"DROP TABLE pgml.{table_name}"#)).unwrap(),
         _ => (),
     }
 }
diff --git a/pgml-extension/tests/test.sql b/pgml-extension/tests/test.sql
@@ -4,7 +4,7 @@
 --- Usage:
 ---
 ---   $ cargo pgx run --release
----   $ psql -h localhost -p 28813 -d pgml -f tests/test.sql -P pager
+---   $ psql -h localhost -p 28815 -d pgml -f tests/test.sql -P pager
 ---
 \set ON_ERROR_STOP true
 \timing on

Original file line number	Diff line number	Diff line change
`@@ -98,7 +98,7 @@ fn drop_table_if_exists(table_name: &str) {`
`98`	`98`	`(PgBuiltInOids::TEXTOID.oid(), table_name.clone().into_datum())`
`99`	`99`	`]).unwrap().unwrap();`
`100`	`100`	`match table_count {`
`101`		`- 1 => Spi::run(&format!(r#"DROP TABLE {table_name}"#)).unwrap(),`
	`101`	`+ 1 => Spi::run(&format!(r#"DROP TABLE pgml.{table_name}"#)).unwrap(),`
`102`	`102`	`_ => (),`
`103`	`103`	`}`
`104`	`104`	`}`