This commit is contained in:
Filippo Olivo
2024-11-27 19:30:12 +01:00
committed by Nicola Demo
parent b52112e448
commit 6da74cadd5
7 changed files with 9 additions and 14 deletions

View File

@@ -134,16 +134,18 @@ class SolverInterface(lightning.pytorch.LightningModule, metaclass=ABCMeta):
return super().on_train_start()
def _check_solver_consistency(self, problem):
"""
TODO
"""
pass
#TODO : Implement this method for the conditions
'''
for _, condition in problem.conditions.items():
if not set(condition.condition_type).issubset(
set(self.accepted_condition_types)):
raise ValueError(
f'{self.__name__} dose not support condition '
f'{condition.condition_type}')
'''
@staticmethod
def get_batch_size(batch):
# Assuming batch is your custom Batch object