🔇 Do not raise deprecation warnings for execute as it's automatically used internally...

* 🔇 Do not raise deprecation warnings for execute as it's automatically used internally

* ✅ Tweak tests to not use deprecated query

diff --git a/sqlmodel/orm/session.py b/sqlmodel/orm/session.py
index 6050d5fbc111a0a71b9953be04a8b00294d491ab..e404bb137d794c5b354c1ef7bebbbf85704372ac 100644 (file)
--- a/sqlmodel/orm/session.py
+++ b/sqlmodel/orm/session.py
@@ -95,7 +95,8 @@ class Session(_Session):
         ```Python
         heroes = session.exec(select(Hero)).all()
         ```
-        """
+        """,
+        category=None,
     )
     def execute(  # type: ignore
         self,

diff --git a/tests/test_main.py b/tests/test_main.py
index 72465cda33015762ae5c298c3531a4a7c9e6886d..bdbcdeb76dfa9b50034c89a6f3f32cb10152a593 100644 (file)
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -3,7 +3,7 @@ from typing import List, Optional
 import pytest\r
 from sqlalchemy.exc import IntegrityError\r
 from sqlalchemy.orm import RelationshipProperty\r
-from sqlmodel import Field, Relationship, Session, SQLModel, create_engine\r
+from sqlmodel import Field, Relationship, Session, SQLModel, create_engine, select\r
 \r
 \r
 def test_should_allow_duplicate_row_if_unique_constraint_is_not_passed(clear_sqlmodel):\r
@@ -31,7 +31,7 @@ def test_should_allow_duplicate_row_if_unique_constraint_is_not_passed(clear_sql
         session.refresh(hero_2)\r
 \r
     with Session(engine) as session:\r
-        heroes = session.query(Hero).all()\r
+        heroes = session.exec(select(Hero)).all()\r
         assert len(heroes) == 2\r
         assert heroes[0].name == heroes[1].name\r
 \r
@@ -61,7 +61,7 @@ def test_should_allow_duplicate_row_if_unique_constraint_is_false(clear_sqlmodel
         session.refresh(hero_2)\r
 \r
     with Session(engine) as session:\r
-        heroes = session.query(Hero).all()\r
+        heroes = session.exec(select(Hero)).all()\r
         assert len(heroes) == 2\r
         assert heroes[0].name == heroes[1].name\r
 \r