Skip to content

Commit f073503

Browse files
authored
fix(textarena): use relative import in rewards
Use a package-relative models import in textarena rewards to avoid import resolution failures.
1 parent d53adee commit f073503

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

envs/textarena_env/rewards.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,7 @@
55
import re
66
from typing import Dict, List, Protocol, Tuple
77

8-
try:
9-
from textarena_env.models import TextArenaAction, TextArenaObservation
10-
except ImportError:
11-
from models import TextArenaAction, TextArenaObservation
8+
from .models import TextArenaAction, TextArenaObservation
129

1310

1411
class RewardProvider(Protocol):

0 commit comments

Comments
 (0)