artificialbrains-sdk 0.1.2__tar.gz → 0.1.4__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (20) hide show
  1. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/PKG-INFO +1 -4
  2. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/README.md +0 -3
  3. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/run_session.py +60 -1
  4. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/artificialbrains_sdk.egg-info/PKG-INFO +1 -4
  5. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/pyproject.toml +1 -1
  6. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/__init__.py +0 -0
  7. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/client.py +0 -0
  8. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/contract_scaffold.py +0 -0
  9. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/endpoints.py +0 -0
  10. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/input_streamer.py +0 -0
  11. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/plugins/__init__.py +0 -0
  12. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/plugins/decoder.py +0 -0
  13. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/plugins/deviation.py +0 -0
  14. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/plugins/reward.py +0 -0
  15. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/ab_sdk/robot_loop.py +0 -0
  16. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/artificialbrains_sdk.egg-info/SOURCES.txt +0 -0
  17. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/artificialbrains_sdk.egg-info/dependency_links.txt +0 -0
  18. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/artificialbrains_sdk.egg-info/requires.txt +0 -0
  19. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/artificialbrains_sdk.egg-info/top_level.txt +0 -0
  20. {artificialbrains_sdk-0.1.2 → artificialbrains_sdk-0.1.4}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: artificialbrains_sdk
3
- Version: 0.1.2
3
+ Version: 0.1.4
4
4
  Summary: ArtificialBrains Python SDK
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: License :: Other/Proprietary License
@@ -88,9 +88,6 @@ python-socketio==5.16.0
88
88
  httpx==0.28.1
89
89
  ```
90
90
 
91
- After installing dependencies you can include this SDK in your
92
- project by copying the `ab_sdk` directory into your source tree or
93
- adding it to your Python path.
94
91
 
95
92
  ## Quick start
96
93
 
@@ -74,9 +74,6 @@ python-socketio==5.16.0
74
74
  httpx==0.28.1
75
75
  ```
76
76
 
77
- After installing dependencies you can include this SDK in your
78
- project by copying the `ab_sdk` directory into your source tree or
79
- adding it to your Python path.
80
77
 
81
78
  ## Quick start
82
79
 
@@ -20,7 +20,7 @@ appropriate synchronization in your handlers.
20
20
  """
21
21
 
22
22
  from __future__ import annotations
23
-
23
+ import struct
24
24
  import logging
25
25
  import threading
26
26
  import time
@@ -198,6 +198,64 @@ class RunSession:
198
198
  }
199
199
  logger.debug("Sending input chunk: %s", {k: payload[k] for k in payload if k != "data"})
200
200
  self.socket.emit("io:chunk", payload, namespace=self.namespace)
201
+
202
+ def send_proprioception_ticks(
203
+ self,
204
+ input_id: str,
205
+ raw_tick: float,
206
+ prev_raw_tick: Optional[float] = None,
207
+ vel_tick_s: Optional[float] = None,
208
+ *,
209
+ cycle: Optional[int] = None,
210
+ seq: Optional[int] = None,
211
+ t: Optional[float] = None,
212
+ lo: Optional[float] = None,
213
+ hi: Optional[float] = None,
214
+ meta: Optional[Dict[str, Any]] = None,
215
+ ) -> None:
216
+ """Send a per-joint proprioception sample as RAW ticks (+ optional velocity).
217
+
218
+ Payload format:
219
+ fmt="rawtick_f32x3"
220
+ data=3x float32 little-endian: [raw_tick, prev_raw_tick, vel_tick_s]
221
+
222
+ Notes:
223
+ - We include cycle in meta for debugging/traceability.
224
+ - For non-Feedback chunks, the server uses its own brainCycle; the cycle
225
+ value is still useful for logs and alignment.
226
+ """
227
+ if prev_raw_tick is None:
228
+ prev_raw_tick = float(raw_tick)
229
+ if vel_tick_s is None:
230
+ vel_tick_s = 0.0
231
+
232
+ payload = struct.pack("<fff", float(raw_tick), float(prev_raw_tick), float(vel_tick_s))
233
+
234
+ m = dict(meta or {})
235
+ if cycle is not None:
236
+ m.setdefault("cycle", int(cycle))
237
+ if lo is not None:
238
+ m.setdefault("lo", float(lo))
239
+ if hi is not None:
240
+ m.setdefault("hi", float(hi))
241
+ m.setdefault("rawTick", float(raw_tick))
242
+ m.setdefault("prevRawTick", float(prev_raw_tick))
243
+ m.setdefault("velTickPerS", float(vel_tick_s))
244
+
245
+ if seq is None:
246
+ seq = int(cycle) if cycle is not None else 0
247
+ if t is None:
248
+ t = time.time()
249
+
250
+ self.send_input_chunk(
251
+ input_id=str(input_id),
252
+ kind="Proprioception",
253
+ seq=int(seq),
254
+ t=float(t),
255
+ fmt="rawtick_f32x3",
256
+ meta=m,
257
+ data=payload,
258
+ )
201
259
 
202
260
  def send_feedback_raster(self, input_id: str, raster: Iterable[float],
203
261
  cycle: int) -> None:
@@ -228,6 +286,7 @@ class RunSession:
228
286
  t=time.time(), fmt="raster_f32",
229
287
  meta=meta, data=data_bytes)
230
288
 
289
+
231
290
  def send_reward(self, global_reward: float, by_layer: Dict[str, float],
232
291
  cycle: int) -> None:
233
292
  """Send reward information to the server.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: artificialbrains_sdk
3
- Version: 0.1.2
3
+ Version: 0.1.4
4
4
  Summary: ArtificialBrains Python SDK
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: License :: Other/Proprietary License
@@ -88,9 +88,6 @@ python-socketio==5.16.0
88
88
  httpx==0.28.1
89
89
  ```
90
90
 
91
- After installing dependencies you can include this SDK in your
92
- project by copying the `ab_sdk` directory into your source tree or
93
- adding it to your Python path.
94
91
 
95
92
  ## Quick start
96
93
 
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "artificialbrains_sdk"
7
- version = "0.1.2"
7
+ version = "0.1.4"
8
8
  description = "ArtificialBrains Python SDK"
9
9
  readme = "README.md"
10
10
  requires-python = ">=3.8"