From 2170ff1c23346dd3323273de16eefced430c4e2e Mon Sep 17 00:00:00 2001 From: Aayam Date: Fri, 7 Feb 2025 07:42:03 -0800 Subject: [PATCH] pre commit run changes --- reasoning_gym/algebra/complex_arithmetic.py | 2 +- tests/test_tsumego.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/algebra/complex_arithmetic.py b/reasoning_gym/algebra/complex_arithmetic.py index 80124447..7c749eaa 100644 --- a/reasoning_gym/algebra/complex_arithmetic.py +++ b/reasoning_gym/algebra/complex_arithmetic.py @@ -72,7 +72,7 @@ class ComplexArithmeticDataset(ProceduralDataset): # For other operations, generate numbers normally a = self._generate_complex(rng) b = self._generate_complex(rng) - + # Calculate result if op == "+": result = a + b diff --git a/tests/test_tsumego.py b/tests/test_tsumego.py index 86ac203f..c1d8d7b6 100644 --- a/tests/test_tsumego.py +++ b/tests/test_tsumego.py @@ -1,8 +1,9 @@ """Tests for Ttsumego problem generation""" -import pytest from random import Random +import pytest + from reasoning_gym.games.tsumego import TsumegoConfig, TsumegoDataset