From 36e0123dd89b515e23f72cbd6c7199fb919d4eb3 Mon Sep 17 00:00:00 2001
From: LeoQuentin <leo.quentin.thorbjornsen.bakholt@nmbu.no>
Date: Sat, 10 Feb 2024 18:06:14 +0100
Subject: [PATCH] fixed import stuff

---
 pyproject.toml                                     | 14 ++++++++------
 .../agents/policy_gradient_agent.py                |  2 +-
 .../agents/value_deep_q_agent.py                   |  2 +-
 3 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index c8b0a9a..f173869 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,17 +1,19 @@
 [tool.poetry]
-name = "special-syllabus"
+name = "ML_Project"
 version = "0.1.0"
-description = "Leo and Hallvard's Special Syllabus about RL and transformers"
+description = "Hallvard and Leo's machine learning endeavor"
 authors = ["Your Name <you@example.com>"]
-license = "n"
+license = "MIT"
 readme = "README.md"
-packages = [{include = "reinforcement_learning", from = "src"}]
+packages = [
+    { include = "reinforcement_learning", from = "src" },
+    { include = "transformers", from = "src" }
+]
 
 [tool.poetry.dependencies]
 python = "^3.10"
 jupyter = "^1.0.0"
 
-
 [build-system]
-requires = ["poetry-core"]
+requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
diff --git a/src/reinforcement_learning/agents/policy_gradient_agent.py b/src/reinforcement_learning/agents/policy_gradient_agent.py
index d497c84..a94a982 100644
--- a/src/reinforcement_learning/agents/policy_gradient_agent.py
+++ b/src/reinforcement_learning/agents/policy_gradient_agent.py
@@ -1,6 +1,6 @@
 import numpy as np
 import torch
-from base_agent import Agent
+from reinforcement_learning.agents import Agent
 
 
 class PolicyGradientAgent(Agent):
diff --git a/src/reinforcement_learning/agents/value_deep_q_agent.py b/src/reinforcement_learning/agents/value_deep_q_agent.py
index 7a2375e..8c4b9b3 100644
--- a/src/reinforcement_learning/agents/value_deep_q_agent.py
+++ b/src/reinforcement_learning/agents/value_deep_q_agent.py
@@ -2,7 +2,7 @@
 import random
 import numpy as np
 import torch
-from reinforcement_learning.agents.base_agent import Agent
+from reinforcement_learning.agents import Agent
 
 
 class ValueDeepQAgent(Agent):