Added train, val, test selctor

backend/alembic/versions/3afa46612906_add_dataset_column_to_signvideo.py

@@ -0,0 +1,43 @@
+"""add dataset column to signvideo
+
+Revision ID: 3afa46612906
+Revises: 2d2d4523082b
+Create Date: 2023-03-22 20:54:32.035246
+
+"""
+import sqlalchemy as sa
+import sqlmodel
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = '3afa46612906'
+down_revision = '2d2d4523082b'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('sign', schema=None) as batch_op:
+        batch_op.alter_column('category_id',
+               existing_type=sa.INTEGER(),
+               nullable=False)
+
+    with op.batch_alter_table('signvideo', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('dataset', sqlmodel.sql.sqltypes.AutoString(), nullable=False, server_default='train'))
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('signvideo', schema=None) as batch_op:
+        batch_op.drop_column('dataset')
+
+    with op.batch_alter_table('sign', schema=None) as batch_op:
+        batch_op.alter_column('category_id',
+               existing_type=sa.INTEGER(),
+               nullable=True)
+
+    # ### end Alembic commands ###