From bc4094ea8c4c9bc8b254dcdcd6c67c21ac6b5abe Mon Sep 17 00:00:00 2001
From: Tom Weber <tom@weber.codes>
Date: Mon, 7 Nov 2022 12:52:27 +0100
Subject: [PATCH] rename

---
 src/drama_generator/generate.py | 4 ++--
 src/drama_generator/model.py    | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/drama_generator/generate.py b/src/drama_generator/generate.py
index 7cd608c..d8a0224 100644
--- a/src/drama_generator/generate.py
+++ b/src/drama_generator/generate.py
@@ -3,8 +3,8 @@
 import re
 import os
 import torch
-from dotd.model import GRU
-from dotd.data import get_shakespeare, preprocess
+from drama_generator.model import GRU
+from drama_generator.data import get_shakespeare, preprocess
 
 
 def get_header():
diff --git a/src/drama_generator/model.py b/src/drama_generator/model.py
index d61fbd1..1148dfb 100644
--- a/src/drama_generator/model.py
+++ b/src/drama_generator/model.py
@@ -5,7 +5,7 @@ import click
 from torch.utils.data import DataLoader
 from tqdm import tqdm  # type: ignore
 import numpy as np
-from dotd.data import ShakespeareDataset, preprocess, get_shakespeare
+from drama_generator.data import ShakespeareDataset, preprocess, get_shakespeare
 
 
 class GRU(torch.nn.Module):