Minor adjustments to Readme

Changelog.md

@@ -2,6 +2,11 @@
 
 ## Versioning
 
+### v0.6.1
+
+* Cleanup in SFNO code
+* Readme should now render correctly in PyPI
+
 ### v0.6.0
 
 * Added SFNO example

README.md

@@ -29,7 +29,7 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 -->
 
 <p align="center">
-    <img src="./images/logo/logo.png"  width="568">
+    <img src="https://github.com/NVIDIA/torch-harmonics/blob/main/images/logo/logo.png"  width="568">
 </p>
 
 <!-- # torch-harmonics: differentiable harmonic transforms -->
@@ -45,9 +45,9 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 <table border="0" cellspacing="0" cellpadding="0">
     <tr>
-        <td><img src="./images/sfno.gif"  width="240"></td>
-        <td><img src="./images/zonal_jet.gif"  width="240"></td>
-        <td><img src="./images/allen-cahn.gif"  width="240"></td>
+        <td><img src="https://github.com/NVIDIA/torch-harmonics/blob/main/images/sfno.gif"  width="240"></td>
+        <td><img src="https://github.com/NVIDIA/torch-harmonics/blob/main/images/zonal_jet.gif"  width="240"></td>
+        <td><img src="https://github.com/NVIDIA/torch-harmonics/blob/main/images/allen-cahn.gif"  width="240"></td>
     </tr>
 <!--     <tr>
         <td style="text-align:center; border-style : hidden!important;">Shallow Water Eqns.</td>

torch_harmonics/examples/sfno/utils/pde_dataset.py

@@ -34,12 +34,7 @@ import os
 
 from math import ceil
 
-import sys
-sys.path.append(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))), "torch_harmonics"))
-sys.path.append(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))), "examples"))
-
-
-from shallow_water_equations import ShallowWaterSolver
+from ...shallow_water_equations import ShallowWaterSolver
 
 class PdeDataset(torch.utils.data.Dataset):
     """Custom Dataset class for PDE training data"""