Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
vmoens committed Mar 18, 2024
1 parent 9f60429 commit b9ad0e2
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions test/test_helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@

import argparse
import dataclasses
import pathlib
import sys

from time import sleep
Expand Down Expand Up @@ -68,12 +69,14 @@

@pytest.fixture
def dreamer_constructor_fixture():
import os

# we hack the env constructor
import sys

sys.path.append(os.path.dirname(__file__) + "/../examples/dreamer/")
sys.path.append(
str(pathlib.Path(__file__).parent.parent / "sota-implementations" / "dreamer")
)

from dreamer_utils import transformed_env_constructor

yield transformed_env_constructor
Expand Down

0 comments on commit b9ad0e2

Please sign in to comment.