NSAPH-Projects / topological-equivariant-networks

E(n)-Equivariant Topological Neural Networks
MIT License
24 stars 1 forks source link

Refactor import/running structure #59

Closed mauriciogtec closed 3 months ago

mauriciogtec commented 5 months ago

Import should be fixed so that code runs from the root folder (it could be convenient to rename src -> etnn).

This is preferable because the etnn/ module becomes importable, and it can be run locally the same way as it would be run after the pip install. For example

import etnn
model = etnn.models.ETNN(...)

imports in the main text should also be from root (e.g., do not use import models directly).