Expose more bindings to module (#26)

* update __init__ with more bindings * format
nerfstudio-project · Oct 5, 2023 · 0fbda5e · 0fbda5e
1 parent a223e5a
commit 0fbda5e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/diff_rast/__init__.py b/diff_rast/__init__.py
@@ -1,9 +1,21 @@
 from .project_gaussians import ProjectGaussians
 from .rasterize import RasterizeGaussians
+from .bin_and_sort_gaussians import BinAndSortGaussians
+from .compute_cumulative_intersects import ComputeCumulativeIntersects
+from .cov2d_bounds import ComputeCov2dBounds
+from .get_tile_bin_edges import GetTileBinEdges
+from .map_gaussian_to_intersects import MapGaussiansToIntersects
+from .sh import SphericalHarmonics
 from .version import __version__
 
 __all__ = [
     "__version__",
     "ProjectGaussians",
     "RasterizeGaussians",
+    "BinAndSortGaussians",
+    "ComputeCumulativeIntersects",
+    "ComputeCov2dBounds",
+    "GetTileBinEdges",
+    "MapGaussiansToIntersects",
+    "SphericalHarmonics",
 ]