import loom.benchmark

DATASET = 'dd-100-100-0.5'


def test_shuffle():
    loom.benchmark.shuffle(DATASET, profile=None)


def test_infer():
    loom.benchmark.infer(DATASET, profile=None)


def test_checkpoint():
    loom.benchmark.load_checkpoint(DATASET)
    loom.benchmark.infer_checkpoint(DATASET, profile=None)


def test_generate():
    loom.benchmark.generate(profile=None)

import loom.benchmark

DATASET = 'dd-100-100-0.5'


def test_shuffle():
    loom.benchmark.shuffle(DATASET, profile=None)


def test_infer():
    loom.benchmark.infer(DATASET, profile=None)


def test_checkpoint():
    loom.benchmark.load_checkpoint(DATASET, period_sec=1)
    loom.benchmark.infer_checkpoint(DATASET, profile=None)


def test_generate():
    loom.benchmark.generate(profile=None)

Reduce test checkpoint period for faster tests
Reduce test checkpoint period for faster tests

  import loom.benchmark
  
  DATASET = 'dd-100-100-0.5'
  
  
  def test_shuffle():
      loom.benchmark.shuffle(DATASET, profile=None)
  
  
  def test_infer():
      loom.benchmark.infer(DATASET, profile=None)
  
  
  def test_checkpoint():
-     loom.benchmark.load_checkpoint(DATASET)
+     loom.benchmark.load_checkpoint(DATASET, period_sec=1)
      loom.benchmark.infer_checkpoint(DATASET, profile=None)
  
  
  def test_generate():
      loom.benchmark.generate(profile=None)
  
Reduce test checkpoint period for faster tests
## Code Before:
import loom.benchmark

DATASET = 'dd-100-100-0.5'


def test_shuffle():
    loom.benchmark.shuffle(DATASET, profile=None)


def test_infer():
    loom.benchmark.infer(DATASET, profile=None)


def test_checkpoint():
    loom.benchmark.load_checkpoint(DATASET)
    loom.benchmark.infer_checkpoint(DATASET, profile=None)


def test_generate():
    loom.benchmark.generate(profile=None)

## Instruction:
Reduce test checkpoint period for faster tests
## Code After:
import loom.benchmark

DATASET = 'dd-100-100-0.5'


def test_shuffle():
    loom.benchmark.shuffle(DATASET, profile=None)


def test_infer():
    loom.benchmark.infer(DATASET, profile=None)


def test_checkpoint():
    loom.benchmark.load_checkpoint(DATASET, period_sec=1)
    loom.benchmark.infer_checkpoint(DATASET, profile=None)


def test_generate():
    loom.benchmark.generate(profile=None)
