diff --git a/pina/collector.py b/pina/collector.py
index 4a7fcdc..f47b57e 100644
--- a/pina/collector.py
+++ b/pina/collector.py
@@ -18,7 +18,7 @@ class Collector:
         and the problem and initializing the data collections (dictionary where
         data will be stored).
 
-        :param pina.problem.abstract_problem.AbstractProblem problem: The 
+        :param pina.problem.abstract_problem.AbstractProblem problem: The
             problem to collect data from.
         """
         # creating a hook between collector and problem
@@ -81,7 +81,7 @@ class Collector:
         """
         Set the problem connected to the collector.
 
-        :param pina.problem.abstract_problem.AbstractProblem value: The problem 
+        :param pina.problem.abstract_problem.AbstractProblem value: The problem
             to connect to the collector.
         """
 
diff --git a/pina/condition/input_equation_condition.py b/pina/condition/input_equation_condition.py
index 7553906..383eb3e 100644
--- a/pina/condition/input_equation_condition.py
+++ b/pina/condition/input_equation_condition.py
@@ -62,7 +62,7 @@ class InputEquationCondition(ConditionInterface):
         Initialize the InputEquationCondition by storing the input and equation.
 
         :param input: Input data for the condition.
-        :type input: pina.label_tensor.LabelTensor | pina.graph.Graph |  
+        :type input: pina.label_tensor.LabelTensor | pina.graph.Graph |
             list[pina.graph.Graph] | tuple[pina.graph.Graph]
         :param EquationInterface equation: Equation object containing the
             equation function.