updating data processing scripts to output to the correct location
This commit is contained in:
@@ -1 +1 @@
|
||||
from src.data.expert_trajectories import demonstrations, load_experts, process_experts
|
||||
from src.data.expert import demonstrations, load_experts, process_experts
|
||||
@@ -1,7 +1,7 @@
|
||||
import torch
|
||||
from torch.utils.data import Dataset
|
||||
import numpy as np
|
||||
from src.expert_data import load_expert_data
|
||||
from src.data.expert_data import load_expert_data
|
||||
import os
|
||||
opj = os.path.join
|
||||
|
||||
|
||||
@@ -100,7 +100,7 @@ def save_video(env, expert):
|
||||
env.render()
|
||||
env.close()
|
||||
|
||||
def load_experts(expert_files=[]):
|
||||
def load_experts(expert_files):
|
||||
"""
|
||||
Load expert trajectories from files and combine their transitions into a single RB
|
||||
|
||||
|
||||
Reference in New Issue
Block a user