ctheodoris madhavanvvs commited on
Commit
5f07ec8
·
verified ·
1 Parent(s): e2efe2b

fully qualified imports to resolve name-space conflicts (#532)

Browse files

- fully qualified imports to resolve name-space conflicts (caecdb154a416524f4913dde407ee7a5bb9d1d2a)


Co-authored-by: Madhavan Venkatesh <[email protected]>

Files changed (1) hide show
  1. geneformer/mtl_classifier.py +3 -4
geneformer/mtl_classifier.py CHANGED
@@ -29,8 +29,7 @@ Geneformer multi-task cell classifier.
29
  import logging
30
  import os
31
 
32
- from .mtl import eval_utils, utils
33
- from .mtl.train import run_manual_tuning, run_optuna_study
34
 
35
  logger = logging.getLogger(__name__)
36
 
@@ -309,7 +308,7 @@ class MTLClassifier:
309
  self.config["manual_hyperparameters"] = self.manual_hyperparameters
310
  self.config["use_manual_hyperparameters"] = True
311
 
312
- run_manual_tuning(self.config)
313
 
314
  def validate_additional_options(self, req_var_dict):
315
  missing_variable = False
@@ -346,7 +345,7 @@ class MTLClassifier:
346
  req_var_dict = dict(zip(required_variable_names, required_variables))
347
  self.validate_additional_options(req_var_dict)
348
 
349
- run_optuna_study(self.config)
350
 
351
  def load_and_evaluate_test_model(
352
  self,
 
29
  import logging
30
  import os
31
 
32
+ from .mtl import eval_utils, utils, train
 
33
 
34
  logger = logging.getLogger(__name__)
35
 
 
308
  self.config["manual_hyperparameters"] = self.manual_hyperparameters
309
  self.config["use_manual_hyperparameters"] = True
310
 
311
+ train.run_manual_tuning(self.config)
312
 
313
  def validate_additional_options(self, req_var_dict):
314
  missing_variable = False
 
345
  req_var_dict = dict(zip(required_variable_names, required_variables))
346
  self.validate_additional_options(req_var_dict)
347
 
348
+ train.run_optuna_study(self.config)
349
 
350
  def load_and_evaluate_test_model(
351
  self,