fix: remove deprecated dj.schema and dj.Di aliases

- Remove `schema` alias for `Schema` (use `dj.Schema` instead of `dj.schema`)
- Remove `Di` alias for `Diagram` (use `dj.Diagram` or `dj.ERD`)
- Update all examples, tests, and docstrings

Co-Authored-By: Claude Opus 4.5 <[email protected]>

Author: dimitri-yatsenko

src/datajoint/__init__.py

@@ -24,7 +24,6 @@
     "conn",
     "Connection",
     "Schema",
-    "schema",
     "VirtualModule",
     "virtual_schema",
     "list_schemas",
@@ -40,7 +39,6 @@
     "Top",
     "U",
     "Diagram",
-    "Di",
     "ERD",
     "kill",
     "MatCell",
@@ -89,8 +87,6 @@
 from .user_tables import Computed, Imported, Lookup, Manual, Part
 from .version import __version__
 
-schema = Schema  # Alias for Schema
-
 # =============================================================================
 # Lazy imports — heavy dependencies loaded on first access
 # =============================================================================
@@ -100,7 +96,6 @@
 _lazy_modules = {
     # Diagram imports networkx and matplotlib
     "Diagram": (".diagram", "Diagram"),
-    "Di": (".diagram", "Diagram"),
     "ERD": (".diagram", "Diagram"),
     "diagram": (".diagram", None),  # Return the module itself
     # kill imports pymysql via connection

src/datajoint/connection.py

@@ -299,7 +299,7 @@ def __enter__(self) -> "Connection":
         Examples
         --------
         >>> with dj.Connection(host, user, password) as conn:
-        ...     schema = dj.schema('my_schema', connection=conn)
+        ...     schema = dj.Schema('my_schema', connection=conn)
         ...     # perform operations
         ... # connection automatically closed
         """

src/datajoint/migrate.py

@@ -90,7 +90,7 @@ def analyze_columns(schema: Schema) -> dict:
     --------
     >>> import datajoint as dj
     >>> from datajoint.migrate import analyze_columns
-    >>> schema = dj.schema('my_database')
+    >>> schema = dj.Schema('my_database')
     >>> result = analyze_columns(schema)
     >>> for col in result['needs_migration']:
     ...     print(f"{col['table']}.{col['column']}: {col['native_type']} → {col['core_type']}")
@@ -315,7 +315,7 @@ def analyze_blob_columns(schema: Schema) -> list[dict]:
     Examples
     --------
     >>> import datajoint as dj
-    >>> schema = dj.schema('my_database')
+    >>> schema = dj.Schema('my_database')
     >>> columns = dj.migrate.analyze_blob_columns(schema)
     >>> for col in columns:
     ...     if col['needs_migration']:

tests/integration/test_codecs.py

@@ -30,7 +30,7 @@ def schema_codec(
     dj.config["stores"] = {"repo-s3": dict(s3_creds, protocol="s3", location="codecs/repo", stage=str(tmpdir))}
     # Codecs are auto-registered via __init_subclass__ in schema_codecs
     context = {**schema_codecs.LOCALS_CODECS}
-    schema = dj.schema(schema_name, context=context, connection=connection_test)
+    schema = dj.Schema(schema_name, context=context, connection=connection_test)
     schema(schema_codecs.Connectivity)
     schema(schema_codecs.Layout)
     yield schema

tests/integration/test_npy_codec.py

@@ -56,7 +56,7 @@ def schema_npy(connection_test, s3_creds, tmpdir, schema_name, mock_stores):
     """Create schema with NpyCodec tables."""
     # mock_stores fixture sets up object_storage.stores with repo-s3, etc.
     context = dict(LOCALS_NPY)
-    schema = dj.schema(schema_name, context=context, connection=connection_test)
+    schema = dj.Schema(schema_name, context=context, connection=connection_test)
     schema(Recording)
     schema(MultiArray)
     yield schema

tests/integration/test_schema.py

@@ -69,7 +69,7 @@ def test_schema_list(schema_any):
 @pytest.mark.requires_mysql
 def test_drop_unauthorized(connection_test):
     """Test that dropping information_schema raises AccessError."""
-    info_schema = dj.schema("information_schema", connection=connection_test)
+    info_schema = dj.Schema("information_schema", connection=connection_test)
     with pytest.raises(dj.errors.AccessError):
         info_schema.drop()
 

tests/unit/test_lazy_imports.py

@@ -89,8 +89,7 @@ def test_diagram_aliases():
 
     import datajoint as dj
 
-    # All aliases should resolve to the same class
-    assert dj.Diagram is dj.Di
+    # ERD alias should resolve to Diagram
     assert dj.Diagram is dj.ERD