calculate_ctf support Tensors as input

teamtomo / libtilt

Image processing for cryo-electron microscopy in PyTorch

Other

14 stars 5 forks source link

calculate_ctf support Tensors as input #49

Closed rsanchezgarc closed 11 months ago

rsanchezgarc commented 11 months ago

calculate_ctf (https://github.com/teamtomo/libtilt/blob/65efd4fbdd262212561651d30d8320a68388623c/src/libtilt/ctf/ctf_2d.py#L12) seems to support Tensors as inputs, but the type hints suggest it only can handle floats.

alisterburt commented 11 months ago

ooh good catch! Want to submit a PR or should I fix?

rsanchezgarc commented 11 months ago

Happy to do it myself as soon as you give me permissions

alisterburt commented 11 months ago

no permissions required for a PR, fork and PR is always available (and my preferred flow for dev) - saying that I'm adding you anyway so you can merge PRs etc :)

rsanchezgarc commented 11 months ago

Solved in #50