Add basic setup.py

interimit/__init__.py

@@ -0,0 +1 @@
+from interimit.expert_data import generate_expert_data, load_expert_data

interimit/data_utils.py

@@ -2,7 +2,7 @@ import torch
 from torch.utils.data import Dataset, DataLoader
 import numpy as np
 #from torchvision import transforms, utils
-from src.expert_data import load_expert_data
+from interimit.expert_data import load_expert_data
 import os
 opj = os.path.join
 

interimit/nets/deepsets.py

@@ -1,7 +1,7 @@
 import torch
 from torch import nn
 
-from src.nets.util import parse_functional
+from interimit.nets.util import parse_functional
 
 class DeepSetsModule(nn.Module):
     def __init__(self, input_dim, phi_hidden_n, phi_hidden_dim, latent_dim, rho_hidden_n, rho_hidden_dim, output_dim):

interimit/policies/policy.py

@@ -2,7 +2,7 @@
 import torch
 from torch import nn
 
-from src.nets.deepsets import DeepSetsModule, Phi
+from interimit.nets.deepsets import DeepSetsModule, Phi
 
 class Policy:
     pass

requirements.txt

@@ -2,3 +2,5 @@ tikzplotlib
 torch
 pytest
 json5
+gym
+git+https://github.com/sisl/InteractionSimulator

setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='interimit',
+    version='0.0.1',
+    url='https://github.com/sisl/InteractionImitation/',
+    author='Arec Jamgochian',
+    author_email='jamgochian95@gmail.com',
+    description='Imitation Learning on INTERACTION Dataset',
+    packages=find_packages(),
+    install_requires=[
+        'tikzplotlib',
+        'torch',
+        'pytest',
+        'json5',
+        'gym',
+        'intersim @ git+https://github.com/sisl/InteractionSimulator',
+    ],
+)

src/__init__.py

@@ -1 +0,0 @@
-from src.expert_data import generate_expert_data, load_expert_data

tests/nets/test_deepsets.py

@@ -1,6 +1,6 @@
 import torch
 import random
-from src.nets import deepsets as ds
+from interimit.nets import deepsets as ds
 import copy
 
 ds_config = {

tests/policies/test_policies.py

@@ -1,5 +1,5 @@
 import torch
-from src.policies.policy import DeepSetsPolicy
+from interimit.policies.policy import DeepSetsPolicy
 import json5
 
 config_path = "config/networks.json5"