add export_onnx to spherinator.models

Author: BerndDoser

src/spherinator/models/__init__.py

@@ -13,6 +13,7 @@
 from .convolutional_encoder_2d import ConvolutionalEncoder2D
 from .dense_model import DenseModel
 from .embedding_reconstruction import EmbeddingReconstruction
+from .export_onnx import export_onnx
 from .variational_autoencoder import VariationalAutoencoder
 from .weights_provider import WeightsProvider
 from .yaml2model import yaml2model
@@ -29,6 +30,7 @@
     "ConvolutionalEncoder2D",
     "DenseModel",
     "EmbeddingReconstruction",
+    "export_onnx",
     "VariationalAutoencoder",
     "WeightsProvider",
     "yaml2model",

src/spherinator/models/export_onnx.py

@@ -2,7 +2,7 @@
 
 import torch
 
-from spherinator.models import yaml2model
+from .yaml2model import yaml2model
 
 
 def export_onnx(