search

johnhw / differentiable_sorting

Differentiable bitonic sorting
MIT License
138 stars 9 forks source link

ModuleNotFoundError: No module named 'differentiable_sorting.torch' #1

Closed n8xm closed 4 years ago

n8xm commented 4 years ago

I have differentiable_sorting installed (using pip install .), and I also have PyTorch and autoray installed. While from differentiable_sorting import bitonic_matrices, diff_sort, diff_argsort does work with no apparent issue, if I run:

import torch
from differentiable_sorting.torch import bitonic_matrices, diff_sort

Then I get the following error message:

ModuleNotFoundError: No module named 'differentiable_sorting.torch'

Am I doing something wrong here?

johnhw commented 4 years ago

Apologies for the delay! This is fixed on PyPI, I'd forgotten to include the critical setuptools.find_packages() line in setup.py