Update API doc

gegewen · Aug 16, 2023 · da0638f · da0638f
1 parent f11291a
commit da0638f
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/doc/source/modules/api.rst b/doc/source/modules/api.rst
@@ -13,7 +13,12 @@ API reference
 Models
 ======
 
-In :mod:`neuralop.models`, we all neural operator models you can directly use on your applications.
+.. automodule:: neuralop.models
+    :no-members:
+    :no-inherited-members:
+
+
+In :mod:`neuralop.models`, we provide neural operator models you can directly use on your applications.
 
 
 FNO
@@ -83,6 +88,11 @@ U-shaped Neural Operators (U-NO)
 Layers
 ======
 
+.. automodule:: neuralop.layers
+    :no-members:
+    :no-inherited-members:
+
+
 In addition to the full architectures, we also provide 
 in :mod:`neuralop.layers` building blocks,
 in the form of PyTorch layers, that you can use to build your own models: