invertedai_simulate.dataset package#

Submodules#

invertedai_simulate.dataset.db module#

class invertedai_simulate.dataset.db.CacheDataDB(sql_file)[source]#

Bases: object

A class holds a sqlite engine session that is able to read from and write to the passed sql_file

count_all() → int[source]#
insert_cached_object(items: List[Tuple[bytes, bytes, bytes]])[source]#
load_data_by_idx(idx: int) → invertedai_simulate.dataset.models.DataLoader[source]#
select_all(limit=1000, offset=0) → List[invertedai_simulate.dataset.models.DataLoader][source]#

invertedai_simulate.dataset.env_dataset module#

class invertedai_simulate.dataset.env_dataset.IAIEnvDataset(config, scenario_name, world_parameters=None, vehicle_physics=None, scenario_parameters=None, sensors=None, env: Optional[invertedai_simulate.interface.IAIEnv] = None)[source]#

Bases: torch.utils.data.dataset.IterableDataset

static add_config(parser: argparse.ArgumentParser) → None[source]#
static build_data_from_queried_result(result: invertedai_simulate.dataset.models.DataLoader) → Dict[source]#
close()[source]#
static get_next_action(info)[source]#
pre_load_data(limit=None)[source]#
set_env_scenario(scenario_name, world_parameters=None, vehicle_physics=None, scenario_parameters=None, sensors=None)[source]#

invertedai_simulate.dataset.models module#

class invertedai_simulate.dataset.models.DataLoader(**kwargs)[source]#

Bases: sqlalchemy.orm.decl_api.Base

IDX#
INFO#
OBS#
REWARDS#

Module contents#