diff --git a/validator/app/src/compute_horde_validator/validator/allowance/tasks.py b/validator/app/src/compute_horde_validator/validator/allowance/tasks.py index f817f88df..30b3950a6 100644 --- a/validator/app/src/compute_horde_validator/validator/allowance/tasks.py +++ b/validator/app/src/compute_horde_validator/validator/allowance/tasks.py @@ -21,6 +21,7 @@ AllowanceMinerManifest, Block, BlockAllowance, + Neuron, SystemEvent, ) @@ -155,6 +156,8 @@ def evict_old_data(): logger.info(f"Removed {removed} AllowanceMinerManifests") removed, _ = Block.objects.filter(block_number__lte=block_number).delete() logger.info(f"Removed {removed} Blocks") + removed, _ = Neuron.objects.filter(block__lte=block_number).delete() + logger.info(f"Removed {removed} Neurons") removed, _ = AllowanceBooking.objects.filter( ~Q( id__in=Subquery( diff --git a/validator/app/src/compute_horde_validator/validator/allowance/tests/test_everything.py b/validator/app/src/compute_horde_validator/validator/allowance/tests/test_everything.py index 1cb6ab2eb..0e20d232a 100644 --- a/validator/app/src/compute_horde_validator/validator/allowance/tests/test_everything.py +++ b/validator/app/src/compute_horde_validator/validator/allowance/tests/test_everything.py @@ -10,7 +10,7 @@ from compute_horde_core.executor_class import ExecutorClass from freezegun import freeze_time -from ...models import AllowanceBooking, AllowanceMinerManifest, Block, BlockAllowance +from ...models import AllowanceBooking, AllowanceMinerManifest, Block, BlockAllowance, Neuron from .. import tasks from ..default import allowance from ..metrics import ( @@ -492,6 +492,7 @@ def test_complete(caplog, configure_logs): assert BlockAllowance.objects.count() == 8402380 assert AllowanceMinerManifest.objects.count() == 7655 assert Block.objects.count() == 1101 + assert set(Neuron.objects.values_list("block", flat=True)) == set(range(1000, 1101)) assert AllowanceBooking.objects.count() == 1 with set_block_number(2906): @@ -500,6 +501,7 @@ def test_complete(caplog, configure_logs): assert BlockAllowance.objects.count() == 2711500 assert AllowanceMinerManifest.objects.count() == 7655 assert Block.objects.count() == 360 + assert set(Neuron.objects.values_list("block", flat=True)) == set(range(1000, 1101)) assert AllowanceBooking.objects.count() == 1 with set_block_number(4000): @@ -508,6 +510,7 @@ def test_complete(caplog, configure_logs): assert BlockAllowance.objects.count() == 0 assert AllowanceMinerManifest.objects.count() == 0 assert Block.objects.count() == 0 + assert Neuron.objects.count() == 0 assert AllowanceBooking.objects.count() == 0