fix procedural generation

OpenZeppelin · Jun 29, 2023 · 15799c1 · 15799c1
1 parent 7940c68
commit 15799c1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/scripts/generate/templates/Checkpoints.js b/scripts/generate/templates/Checkpoints.js
@@ -5,8 +5,8 @@ const { OPTS } = require('./Checkpoints.opts.js');
 const header = `\
 pragma solidity ^0.8.19;
 
-import "../math/Math.sol";
-import "../math/SafeCast.sol";
+import {Math} from "../math/Math.sol";
+import {SafeCast} from "../math/SafeCast.sol";
 
 /**
  * @dev This library defines the \`History\` struct, for checkpointing values as they change at different points in
@@ -21,7 +21,7 @@ import "../math/SafeCast.sol";
 
 const errors = `\
     /**
-     * @dev A value was attempted to be inserted on a past checkpoint. 
+     * @dev A value was attempted to be inserted on a past checkpoint.
      */
     error CheckpointUnorderedInsertion();
 `;

diff --git a/scripts/generate/templates/Checkpoints.t.js b/scripts/generate/templates/Checkpoints.t.js
@@ -6,9 +6,9 @@ const { OPTS } = require('./Checkpoints.opts.js');
 const header = `\
 pragma solidity ^0.8.19;
 
-import "forge-std/Test.sol";
-import "../../../contracts/utils/math/SafeCast.sol";
-import "../../../contracts/utils/structs/Checkpoints.sol";
+import {Test} from "forge-std/Test.sol";
+import {SafeCast} from "../../../contracts/utils/math/SafeCast.sol";
+import {Checkpoints} from "../../../contracts/utils/structs/Checkpoints.sol";
 `;
 
 /* eslint-disable max-len */

diff --git a/scripts/generate/templates/EnumerableMap.js b/scripts/generate/templates/EnumerableMap.js
@@ -12,7 +12,7 @@ const TYPES = [
 const header = `\
 pragma solidity ^0.8.19;
 
-import "./EnumerableSet.sol";
+import {EnumerableSet} from "./EnumerableSet.sol";
 
 /**
  * @dev Library for managing an enumerable variant of Solidity's