From b7cd677a20b542d3bbf854e1b374e2155b048c14 Mon Sep 17 00:00:00 2001
From: Raffi Khatchadourian <khatchad@hunter.cuny.edu>
Date: Thu, 1 Aug 2024 16:18:08 -0400
Subject: [PATCH] Actually use `pickle_dir` `command-line argument.

Previously, the directory was hard-coded and the command-line parameter wasn't
referenced.
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 767586b..29e08c4 100644
--- a/train.py
+++ b/train.py
@@ -41,7 +41,7 @@
 
 
 # load data
-dataset = Data('dataset/processed')
+dataset = Data(pickle_dir)
 print(dataset)