From 229716ffbd50b14b745a1b048d6737a4a60e2bb3 Mon Sep 17 00:00:00 2001
From: Nitesh Kumar <166297874+niteshg97@users.noreply.github.com>
Date: Sun, 24 May 2026 09:42:45 +0530
Subject: [PATCH] Add missing `self` parameter to `calculate_additional_loss`.

Added missing self-parameter to calculate additional loss
---
 src/pquant/pruning_methods/fitcompress.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/pquant/pruning_methods/fitcompress.py b/src/pquant/pruning_methods/fitcompress.py
index f2b3b5c..9e5b314 100644
--- a/src/pquant/pruning_methods/fitcompress.py
+++ b/src/pquant/pruning_methods/fitcompress.py
@@ -26,7 +26,7 @@ def get_hard_mask(self, weight=None):
     def pre_epoch_function(self, epoch, total_epochs):
         pass
 
-    def calculate_additional_loss(*args, **kwargs):
+    def calculate_additional_loss(self):
         return 0
 
     def pre_finetune_function(self):