antialias/haunt-fm
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

Fix SQLAlchemy import: Real -> REAL

Browse Source 
SQLAlchemy exports REAL (uppercase), not Real.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Thomas Hallock
2026-02-22 08:59:10 -06:00
parent 179071b817
commit f8c13daa11
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
alembic/versions/001_initial_schema.py
View File

@@ -73,7 +73,7 @@ def upgrade() -> None:
sa.Column("id", sa.BigInteger, primary_key=True), sa.Column("id", sa.BigInteger, primary_key=True),
sa.Column("source_track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False), sa.Column("source_track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False),
sa.Column("target_track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False), sa.Column("target_track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False),
sa.Column("lastfm_match", sa.Real), sa.Column("lastfm_match", sa.REAL),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()), sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
sa.UniqueConstraint("source_track_id", "target_track_id", name="uq_similarity_link"), sa.UniqueConstraint("source_track_id", "target_track_id", name="uq_similarity_link"),
) )
@@ -105,7 +105,7 @@ def upgrade() -> None:
sa.Column("track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False), sa.Column("track_id", sa.BigInteger, sa.ForeignKey("tracks.id"), nullable=False),
sa.Column("position", sa.Integer, nullable=False), sa.Column("position", sa.Integer, nullable=False),
sa.Column("is_known", sa.Boolean, nullable=False), sa.Column("is_known", sa.Boolean, nullable=False),
sa.Column("similarity_score", sa.Real), sa.Column("similarity_score", sa.REAL),
) )

6
src/haunt_fm/models/track.py
View File

@@ -1,7 +1,7 @@
from datetime import datetime from datetime import datetime
from pgvector.sqlalchemy import Vector from pgvector.sqlalchemy import Vector
from sqlalchemy import BigInteger, DateTime, Index, Integer, Real, Text, func from sqlalchemy import REAL, BigInteger, DateTime, Index, Integer, Text, func
from sqlalchemy.dialects.postgresql import JSONB from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.orm import Mapped, mapped_column, relationship
@@ -69,7 +69,7 @@ class SimilarityLink(Base):
id: Mapped[int] = mapped_column(BigInteger, primary_key=True) id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
source_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) source_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
target_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) target_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
lastfm_match: Mapped[float | None] = mapped_column(Real) lastfm_match: Mapped[float | None] = mapped_column(REAL)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
__table_args__ = ( __table_args__ = (
@@ -107,7 +107,7 @@ class PlaylistTrack(Base):
track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) track_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
position: Mapped[int] = mapped_column(Integer, nullable=False) position: Mapped[int] = mapped_column(Integer, nullable=False)
is_known: Mapped[bool] = mapped_column(nullable=False) is_known: Mapped[bool] = mapped_column(nullable=False)
similarity_score: Mapped[float | None] = mapped_column(Real) similarity_score: Mapped[float | None] = mapped_column(REAL)
playlist: Mapped[Playlist] = relationship(back_populates="tracks") playlist: Mapped[Playlist] = relationship(back_populates="tracks")
track: Mapped[Track] = relationship() track: Mapped[Track] = relationship()