From 16bcd7a2c35e5642188f08c1552269b8ad86a57d Mon Sep 17 00:00:00 2001 From: Nelson Torres Date: Thu, 30 Jan 2025 02:14:47 +0000 Subject: [PATCH] fix the input values for acm related to oi --- max_pain.py | 2 -- piker/brokers/deribit/api.py | 9 ++++++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/max_pain.py b/max_pain.py index 34bc9327..5446aa25 100644 --- a/max_pain.py +++ b/max_pain.py @@ -8,8 +8,6 @@ from piker.brokers.deribit.api import ( async def max_pain_daemon( ) -> None: - async with maybe_open_oi_feed() as oi_feed: - print('Im in...') def check_if_complete( oi: dict[str, dict[str, Decimal | None]], diff --git a/piker/brokers/deribit/api.py b/piker/brokers/deribit/api.py index 50f7dde8..bc11337e 100644 --- a/piker/brokers/deribit/api.py +++ b/piker/brokers/deribit/api.py @@ -812,8 +812,8 @@ async def maybe_open_price_feed( async def aio_open_interest_feed_relay( fh: FeedHandler, - instruments: list, oi_by_strikes: dict[str, dict[str, Decimal]], + instruments: list[Symbol], from_trio: asyncio.Queue, to_trio: trio.abc.SendChannel, ) -> None: @@ -915,6 +915,7 @@ async def aio_open_interest_feed_relay( @acm async def open_oi_feed( + instruments: list[Symbol], ) -> to_asyncio.LinkedTaskChannel: expiry_date: str = '6DEC24' instruments: list[Symbol] = [] @@ -946,12 +947,18 @@ async def open_oi_feed( @acm async def maybe_open_oi_feed( + instruments: list[Symbol], ) -> trio.abc.ReceiveStream: # TODO: add a predicate to maybe_open_context feed: to_asyncio.LinkedTaskChannel async with maybe_open_context( acm_func=open_oi_feed, + kwargs={ + 'instruments': instruments + }, + key=f'{instruments[0]}', + ) as (cache_hit, feed): if cache_hit: yield broadcast_receiver(feed, 10)