search

tkipf / c-swm

Contrastive Learning of Structured World Models
MIT License
386 stars 67 forks source link

Add to path in physics.py #8

Open csquires opened 3 years ago

csquires commented 3 years ago

You need to copy the following lines from data_gen/env.py into data_gen/physics.py:

# Get env directory
import sys
from pathlib import Path
if str(Path.cwd()) not in sys.path:
    sys.path.insert(0, str(Path.cwd()))