playing around with build system

.python-version

@@ -1 +0,0 @@
-3.7.7

README.md

@@ -0,0 +1 @@
+# DRAMA GENERATOR 

pyproject.toml

@@ -16,6 +16,10 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
 ]
+dependencies = [
+    "torch >= 1.10.0",
+    "numpy>=1.21.5"
+    ]
 
 [project.urls]
 "repository" = "https://git.weber.codes/tom/drama_generator.git"

src/drama_generator/model.py

@@ -2,7 +2,6 @@
 """ module for training and creating the model """
 import torch
 from torch.utils.data import DataLoader
-from tqdm import tqdm  # type: ignore
 import numpy as np
 from drama_generator.data import ShakespeareDataset, preprocess, get_shakespeare
 
@@ -70,7 +69,7 @@ def train(epochs: int, save_path: str):
         num_workers=2,
         drop_last=True,
     )
-    for e in tqdm(range(epochs)):
+    for e in range(epochs):
         l = train_epoch(gru, loader, optim, loss, device)
         print(f"Epoch: {e}, Loss: {l}")
     torch.save(gru.state_dict(), save_path + f"gru_{epochs+1}epochs.pth")