removed extraneous comments

Author: cxzhang4

NEWS.md

@@ -1,5 +1,6 @@
 # mlr3torch dev
 
+* Added `CallbackSetTB`, which allows logging that can be viewed by TensorBoard.
 * Don't use deprecated `data_formats` anymore
 
 # mlr3torch 0.1.1

R/CallbackSetTB.R

@@ -30,8 +30,6 @@ CallbackSetTB = R6Class("CallbackSetTB",
     },
     #' @description
     #' Logs the training loss and validation measures as TensorFlow events.
-    # TODO: display the appropriate x axis with its label in TensorBoard
-    # relevant when we log different scores at different times
     on_epoch_end = function() {
       if (self$log_train_loss) {
         private$.log_train_loss()
@@ -64,7 +62,6 @@ CallbackSetTB = R6Class("CallbackSetTB",
       private$.log_score("train.", measure_name, train_score)
     },
     .log_train_loss = function() {
-      # TODO: remind ourselves why we wanted to display last_loss and not last_scores_train
       with_logdir(self$path, {
         log_event(train.loss = self$ctx$last_loss)
       })

tests/testthat/test_CallbackSetTB.R

@@ -7,7 +7,6 @@ test_that("autotest", {
 
 # TODO: investigate what's happening when there is only a single epoch (why don't we log anything?)
 test_that("a simple example works", {
-  # using a temp dir
   cb = t_clbk("tb")
 
   task = tsk("iris")
@@ -53,7 +52,6 @@ test_that("a simple example works", {
 })
 
 test_that("eval_freq works", {
-  # using a temp dir
   cb = t_clbk("tb")
 
   task = tsk("iris")