add links

AlgorithmicResearchGroup · Dec 8, 2024 · 00c7b9a · 00c7b9a
1 parent 8e18caf
commit 00c7b9a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -2,6 +2,11 @@
 
 The ML Research Benchmark Baseline Agent is an agentic system designed to serve as a baseline for various AI and machine learning tasks. This agent provides a foundation for comparing and evaluating machine learning research and development tasks that agents can perform.
 
+- [:paperclip: ML Research Benchmark Paper](https://arxiv.org/abs/2410.22553) [![arXiv](https://img.shields.io/badge/arXiv-1234.56789-b31b1b.svg)](https://arxiv.org/abs/1234.56789)
+- [:robot: ML Research Agent](https://github.com/AlgorithmicResearchGroup/ML-Research-Agent)
+- [:white_check_mark: ML Research Tasks](https://github.com/AlgorithmicResearchGroup/ML-Research-Agent-Tasks)
+- [:chart_with_upwards_trend: ML Research Evaluation](https://github.com/AlgorithmicResearchGroup/ML-Research-Agent-Evals)
+
 
 ## Features
 

diff --git a/agent/task.py b/agent/task.py
@@ -7,7 +7,7 @@
 import numpy as np
 import tiktoken
 from datasets import load_dataset  # huggingface datasets
-from agent_tasks import get_task
+from mlrb_agent_tasks import get_task
 
 
 def make_directory(work_dir):