Aswini-Kumar commited on
Commit
bfe686c
·
verified ·
1 Parent(s): ed87722

Sync client.py

Browse files
Files changed (1) hide show
  1. client.py +4 -1
client.py CHANGED
@@ -12,7 +12,10 @@ from openenv.core import EnvClient
12
  from openenv.core.client_types import StepResult
13
  from openenv.core.env_server.types import State
14
 
15
- from .models import DataWranglerAction, DataWranglerObservation
 
 
 
16
 
17
 
18
  class DataWranglerEnv(
 
12
  from openenv.core.client_types import StepResult
13
  from openenv.core.env_server.types import State
14
 
15
+ try:
16
+ from .models import DataWranglerAction, DataWranglerObservation
17
+ except ImportError:
18
+ from models import DataWranglerAction, DataWranglerObservation
19
 
20
 
21
  class DataWranglerEnv(