first commit

README.md

@@ -0,0 +1 @@
+# AIMO

aimo/__init__.py

@@ -0,0 +1,4 @@
+
+from .competition import make_env
+
+__all__ = ['make_env']

code.py

@@ -0,0 +1,16 @@
+
+try:
+    from sympy import *
+
+    def max_yellow_numbers():
+        min_blue = 222
+        max_blue = 888
+        num_blue = max_blue / 2 + 1
+        return num_blue
+
+    result = max_yellow_numbers()
+    print(int(result))
+
+except Exception as e:
+    print(e)
+    print('FAIL')

configs/config.yaml

@@ -0,0 +1,30 @@
+LOGGER: True
+SEED: 42
+DEBUG: False
+QUANT: False
+USE_PAST_KEY: True
+PRIVATE: True
+
+n_repetitions: 19 # Original notebook had 22 but times out :(
+TOTAL_TOKENS: 2048 # if PRIVATE else 512
+
+DEEP: True
+
+code: """Below is a math problem you are to solve (positive numerical answer):
+\"{}\"
+To accomplish this, first determine a sympy-based approach for solving the problem by listing each step to take and what functions need to be called in each step. Be clear so even an idiot can follow your instructions, and remember, your final answer should be positive integer, not an algebraic expression!
+Write the entire script covering all the steps (use comments and document it well) and print the result. After solving the problem, output the final numerical answer within \\boxed{}.
+
+Approach:"""
+
+
+cot: """Below is a math problem you are to solve (positive numerical answer!):
+\"{}\"Analyze this problem and think step by step to come to a solution with programs. After solving the problem, output the final numerical answer within \\boxed{}.\n\n"""
+
+tool_instruction: '\n\nPlease integrate natural language reasoning with programs to solve the above problem, and put your final numerical answer within \\boxed{}.\nNote that the intermediary calculations may be real numbers, but the final numercal answer would always be an integer.'
+
+
+temperature: 0.9
+top_p: 1.0
+
+

data/sample_submission.csv

@@ -0,0 +1,4 @@
+row_id,id,answer
+0,000aaa,0
+1,111bbb,0
+2,222ccc,0

data/test.csv

@@ -0,0 +1,4 @@
+row_id,id,problem
+0,000aaa,What is $1-1$?
+1,111bbb,What is $0\times10$?
+2,222ccc,Solve $4+x=4$ for $x$.

data/train.csv

@@ -0,0 +1,11 @@
+"id","problem","answer"
+"229ee8","Let $k, l > 0$ be parameters. The parabola $y = kx^2 - 2kx + l$ intersects the line $y = 4$ at two points $A$ and $B$. These points are distance 6 apart. What is the sum of the squares of the distances from $A$ and $B$ to the origin?",52
+"246d26","Each of the three-digits numbers $111$ to $999$ is coloured blue or yellow in such a way that the sum of any two (not necessarily different) yellow numbers is equal to a blue number. What is the maximum possible number of yellow numbers there can be?",250
+"2fc4ad","Let the `sparkle' operation on positive integer $n$ consist of calculating the sum of the digits of $n$ and taking its factorial, e.g. the sparkle of 13 is $4! = 24$. A robot starts with a positive integer on a blackboard, then after each second for the rest of eternity, replaces the number on the board with its sparkle. For some `special' numbers, if they're the first number, then eventually every number that appears will be less than 6. How many such special numbers are there with at most 36 digits?",702
+"430b63","What is the minimum value of $5x^2+5y^2-8xy$ when $x$ and $y$ range over all real numbers such that $|x-2y| + |y-2x| = 40$?",800
+"5277ed","There exists a unique increasing geometric sequence of five 2-digit positive integers. What is their sum?",211
+"739bc9","For how many positive integers $m$ does the equation \[\vert \vert x-1 \vert -2 \vert=\frac{m}{100}\] have $4$ distinct solutions?",199
+"82e2a0","Suppose that we roll four 6-sided fair dice with faces numbered 1 to~6. Let $a/b$ be the probability that the highest roll is a 5, where $a$ and $b$ are relatively prime positive integers. Find $a + b$.",185
+"8ee6f3","The points $\left(x, y\right)$ satisfying $((\vert x + y \vert - 10)^2 + ( \vert x - y \vert - 10)^2)((\vert x \vert - 8)^2 + ( \vert y \vert - 8)^2) = 0$ enclose a convex polygon. What is the area of this convex polygon?",320
+"bedda4","Let $ABCD$ be a unit square. Let $P$ be the point on $AB$ such that $|AP| = 1/{20}$ and let $Q$ be the point on $AD$ such that $|AQ| = 1/{24}$. The lines $DP$ and $BQ$ divide the square into four regions. Find the ratio between the areas of the largest region and the smallest region.",480
+"d7e9c9","A function $f: \mathbb N \to \mathbb N$ satisfies the following two conditions for all positive integers $n$:$f(f(f(n)))=8n-7$ and $f(2n)=2f(n)+1$. Calculate $f(100)$.",199

log.txt

@@ -0,0 +1 @@
+Loading the config file...