From 255189b34f7a0d6e2c5e148a1aadfb231fc788b0 Mon Sep 17 00:00:00 2001
From: Andi Gerken <andi.gerken@gmail.com>
Date: Wed, 10 Feb 2021 14:34:17 +0100
Subject: [PATCH] [ci] changed to push artifacts to the artifacts repository
 Updated example, updated url of track format

---
 ci/deploy.py               |  3 +--
 examples/example_readme.py | 25 ++++++++++++++++++-------
 src/robofish/io/file.py    |  4 +++-
 3 files changed, 22 insertions(+), 10 deletions(-)

diff --git a/ci/deploy.py b/ci/deploy.py
index 8ca66d5..6dda82b 100755
--- a/ci/deploy.py
+++ b/ci/deploy.py
@@ -18,8 +18,7 @@ if __name__ == "__main__":
     command += ["-m", "twine", "upload", "dist/*"]
     command += [
         "--repository-url",
-        f"https://git.imp.fu-berlin.de/api/v4/projects/{env['CI_PROJECT_ID']}/packages/pypi",
+        f"https://git.imp.fu-berlin.de/api/v4/projects/{env['ARTIFACTS_REPOSITORY_PROJECT_ID']}/packages/pypi",
     ]
-    # ARTIFACTS_REPOSITORY_PROJECT_ID
 
     check_call(command)
diff --git a/examples/example_readme.py b/examples/example_readme.py
index f006082..a568f8a 100644
--- a/examples/example_readme.py
+++ b/examples/example_readme.py
@@ -1,16 +1,27 @@
 import robofish.io
-import numpy
+import numpy as np
+
+filename = "example.hdf5"
 
 f = robofish.io.File(world_size_cm=[100, 100], frequency_hz=25.0)
 
-# Create a single robot with 40 poses
+# Create a single robot with 20 timesteps
 # (x,y, x_orientation, y_orientation) and specified time points
-f.create_entity(category="robot", name="robot", poses=numpy.zeros((40, 4)))
 
-# Create 2 fishes with 100 poses
-# (x,y, x_orientation, y_orientation) and 40ms timesteps
-f.create_multiple_entities("fish", poses=numpy.zeros((2, 100, 4)))
+f.create_entity(
+    category="robot",
+    name="robot",
+    positions=np.ones((20, 2)) * 10,
+    orientations=np.zeros((20, 1)),
+)
+
+# Create 2 fishes with 20 poses
+poses = np.zeros((2, 20, 4))
+poses[:, :, 3] = 1  # all fishes pointing upwards
+poses[1, :, :2] = -20  # move first fish to x,y -20,-20
+f.create_multiple_entities("fish", poses=poses)
 
 # Show and save the file
 print(f)
-f.save("example.hdf5")
+f.save(filename)
+print(f"Saved to {filename}")
diff --git a/src/robofish/io/file.py b/src/robofish/io/file.py
index 272de38..3267763 100644
--- a/src/robofish/io/file.py
+++ b/src/robofish/io/file.py
@@ -32,7 +32,9 @@ import deprecation
 temp_dir = tempfile.TemporaryDirectory()
 default_format_version = np.array([1, 0], dtype=np.int32)
 
-default_format_url = "https://git.imp.fu-berlin.de/bioroboticslab/robofish/track_format/-/releases/1.0-draft.10"
+default_format_url = (
+    "https://git.imp.fu-berlin.de/bioroboticslab/robofish/track_format/-/releases/1.0"
+)
 
 
 class File(h5py.File):
-- 
GitLab