Using a relative path to reload swarm files

underworldcode · Jul 26, 2021 · cbc9a26 · cbc9a26
1 parent 63e7384
commit cbc9a26
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/docs/examples/07_ShearBandsPureShear.ipynb b/docs/examples/07_ShearBandsPureShear.ipynb
@@ -175,8 +175,10 @@
     "    swarmLayout = uw.swarm.layouts.PerCellSpaceFillerLayout( swarm=swarm, particlesPerCell=parts_per_cell )\n",
     "    swarm.populate_using_layout( layout=swarmLayout )\n",
     "else:\n",
+    "    import os\n",
+    "    dirname = os.path.dirname(__file__)\n",
     "    # load in parallel for deterministic results\n",
-    "    swarm.load(\"input/07_ShearBandsPureShear/particle_coords.h5\")\n",
+    "    swarm.load(dirname+\"/input/07_ShearBandsPureShear/particle_coords.h5\")\n",
     "# create pop control object\n",
     "pop_control = uw.swarm.PopulationControl(swarm, aggressive=True, aggressiveThreshold=0.8, particlesPerCell=parts_per_cell)\n",
     "\n",
@@ -277,8 +279,10 @@
     "    plasticStrain.data[:,0] *= gaussian(swarm.particleCoordinates.data[:,1], 0.0, 0.025) \n",
     "    plasticStrain.data[:,0] *= boundary(swarm.particleCoordinates.data[:,0], minX, maxX, 10.0, 2) \n",
     "else:\n",
+    "    import os\n",
+    "    dirname = os.path.dirname(__file__)\n",
     "    # load in parallel for deterministic results\n",
-    "    plasticStrain.load(\"input/07_ShearBandsPureShear/plastic_strain.h5\")\n",
+    "    plasticStrain.load(dirname+\"/input/07_ShearBandsPureShear/plastic_strain.h5\")\n",
     "\n",
     "deformationVariable.data[:,0] = deformationSwarm.particleCoordinates.data[:,1]"
    ]
@@ -1053,7 +1057,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.3"
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,