fixed import stuff

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"

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):

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):